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