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.
