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