Searched refs:_wrap_backward_function (Results 1 – 1 of 1) sorted by relevance
970 py_backward, _ = self._wrap_backward_function(997 py_backward, recorded_outputs = self._wrap_backward_function(1113 def _wrap_backward_function(self, forward_graph, backward, outputs): member in _TapeGradientFunctions1205 backward_function, to_record = self._wrap_backward_function(