Searched defs:primal (Results 1 – 6 of 6) sorted by relevance
398 def _watch(primal, tangent): argument
91 def _jvp_batch(f, primal, tangents): argument576 def _compute_forwardgrad(primal): argument
130 def _grad_function(primal): argument
1823 def _grad_function(primal): argument
966 def _grad_function(primal): argument
196 def _GradFunction(primal): argument