parameters: (self, op, before)
docstring: Exit the WhileContext for gradient computation.
docstring_summary: Exit the WhileContext for gradient computation.
docstring_tokens: Exit the WhileContext for gradient computation .
function: def ExitGradWhileContext(self, op, before):
      """Exit the WhileContext for gradient computation."""
      grad_state = self._GetGradState(op, before)
      if grad_state:
        grad_state.grad_context.Exit()
function_tokens: def ExitGradWhileContext ( self , op , before ) : grad_state = self . _GetGradState ( op , before ) if grad_state : grad_state . grad_context . Exit ( )