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