Searched refs:def_deriv (Results 1 – 1 of 1) sorted by relevance
265 def def_deriv(prim, deriv): function286 def_deriv(lax.erf_p, lambda x: lax.mul(lax._const(x, 2. / np.sqrt(np.pi)), lax.exp(lax.neg(lax.squa…