Searched refs:retain_variables (Results 1 – 1 of 1) sorted by relevance
779 def _do_backward(self, gradients, retain_variables): argument780 self.retain_variables = retain_variables781 result = super()._do_backward(gradients, retain_variables) # type: ignore[misc]782 if not retain_variables: