Searched refs:rhs_jvp (Results 1 – 1 of 1) sorted by relevance
487 rhs_jvp = lambda g, x, y, **kwargs: prim.bind(x, bcast(g, x), **kwargs) function488 defjvp(prim, lhs_jvp, rhs_jvp)