Searched refs:_jvp_batch (Results 1 – 1 of 1) sorted by relevance
91 def _jvp_batch(f, primal, tangents): function1081 _jvp_batch(f, primals, tangent_batch)[1],