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