Searched refs:_wrap_backward_function (Results 1 – 1 of 1) sorted by relevance
1068 py_backward, _ = self._wrap_backward_function(1095 py_backward, recorded_outputs = self._wrap_backward_function(1211 def _wrap_backward_function(self, forward_graph, backward, outputs): member in _TapeGradientFunctions1303 backward_function, to_record = self._wrap_backward_function(