def solve(a, b): a, b = _promote_arg_dtypes(jnp.asarray(a), jnp.asarray(b)) return lax_linalg._solve(a, b)
def solve(a, b): a, b = _promote_dtypes_inexact(jnp.asarray(a), jnp.asarray(b)) return lax_linalg._solve(a, b)
def solve(a, b): a, b = _promote_arg_dtypes(jnp.asarray(a), jnp.asarray(b)) return lax_linalg._solve(a, b)
def solve(a, b): a, b = _promote_dtypes_inexact(jnp.asarray(a), jnp.asarray(b)) return lax_linalg._solve(a, b)