Searched refs:_wrap_backward_function_with_jvp_backprop (Results 1 – 1 of 1) sorted by relevance
877 forward_wrapper) = self._wrap_backward_function_with_jvp_backprop(991 def _wrap_backward_function_with_jvp_backprop( member in _TapeGradientFunctions