Searched refs:_wrap_forward_function_with_jvps (Results 1 – 1 of 1) sorted by relevance
874 forward_wrapper = self._wrap_forward_function_with_jvps(897 def _wrap_forward_function_with_jvps( member in _TapeGradientFunctions