Searched defs:primal (Results 1 – 6 of 6) sorted by relevance
402 def _watch(primal, tangent): argument
90 def _jvp_batch(f, primal, tangents): argument532 def _compute_forwardgrad(primal): argument
133 def _grad_function(primal): argument
1766 def _grad_function(primal): argument
970 def _grad_function(primal): argument
181 def _GradFunction(primal): argument