def sinc(x): _check_arraylike("sinc", x) x, = _promote_dtypes_inexact(x) eq_zero = lax.eq(x, _lax_const(x, 0)) pi_x = lax.mul(_lax_const(x, np.pi), x) safe_pi_x = _where(eq_zero, _lax_const(x, 1), pi_x) return _where(eq_zero, _sinc_maclaurin(0, pi_x), lax.div(lax.sin(safe_pi_x), safe_pi_x))
def f(x, y): z = lax.mul(x, y) w = lax.sin(z) u = lax_internal._reduce_sum(w, [0]) return (u,)
def f(x): return lax.sin(x) * lax.cos(lax.stop_gradient(x))
def f2(x, y): return lax.sin(x) * lax.cos(y)
def sin(x): x, = _promote_to_result_dtype(onp.sin, x) return lax.sin(x)
def f(x): return lax.cos(lax.sin(x))
def g(x): return lax.sin(lax.sin(x))
def g(x): if x > 0: return lax.sin(x), 3. else: return lax.cos(x), 4.