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