Searched refs:forward_context (Results 1 – 2 of 2) sorted by relevance
299 IsContainingContext(while_ctxt.grad_state.forward_context,309 while_ctxt.grad_state.forward_context is316 input_while_ctxt.grad_state.forward_context is while_ctxt):322 input_ctxt.grad_state.forward_context.grad_state and323 input_ctxt.grad_state.forward_context.grad_state.forward_context is
695 outer_forward_ctxt = outer_grad_state.forward_context753 def forward_context(self): member in GradLoopState883 value, self.forward_context)890 enter_acc = self.forward_context.AddValue(acc)893 swap_enabled = self.forward_context.swap_memory895 if value_ctxt == self.forward_context:897 self.forward_context.Enter()900 self.forward_context.Exit()1094 for y in grad_state.forward_context.loop_enters:2421 if forward_ctxt == grad_ctxt.grad_state.forward_context:[all …]