Esempio n. 1
0
def get_dtypes(*args):
    return [canonicalize_dtype(onp.result_type(arg)) for arg in args]
Esempio n. 2
0
def get_dtypes(*args):
    return [canonicalize_dtype(lax.dtype(arg)) for arg in args]