Searched defs:primal (Results 1 – 9 of 9) sorted by relevance
156 const Tensor& primal, in _make_dual()503 const Tensor& primal, in _make_dual()
22 Tensor _make_dual(const Tensor& primal, const Tensor& tangent, int64_t level) { in _make_dual()
91 def _jvp_batch(f, primal, tangents): argument576 def _compute_forwardgrad(primal): argument
398 def _watch(primal, tangent): argument
130 def _grad_function(primal): argument
1823 def _grad_function(primal): argument
966 def _grad_function(primal): argument
196 def _GradFunction(primal): argument
847 const Tensor& primal, in _make_dual_batching_rule()