Searched refs:_rewrite_forward_and_call_backward (Results 1 – 1 of 1) sorted by relevance
740 def _rewrite_forward_and_call_backward(self, op, *doutputs): member in _DelayedRewriteGradientFunctions791 return self._rewrite_forward_and_call_backward827 return self._rewrite_forward_and_call_backward(call_op, *args)1057 …self._delayed_rewrite_functions._rewrite_forward_and_call_backward) # pylint: disable=protected-a…2087 …return self._delayed_rewrite_functions._rewrite_forward_and_call_backward # pylint: disable=prote…