Searched refs:forward_context (Results 1 – 3 of 3) sorted by relevance
312 IsContainingContext(while_ctxt.grad_state.forward_context,322 while_ctxt.grad_state.forward_context is329 input_while_ctxt.grad_state.forward_context is while_ctxt):335 input_ctxt.grad_state.forward_context.grad_state and336 input_ctxt.grad_state.forward_context.grad_state.forward_context is
156 outer_forward_ctxt = outer_grad_state.forward_context214 def forward_context(self): member in _GradLoopState344 value, self.forward_context)351 enter_acc = self.forward_context.AddValue(acc)354 swap_enabled = self.forward_context.swap_memory356 if value_ctxt == self.forward_context:358 self.forward_context.Enter()361 self.forward_context.Exit()548 for y in grad_state.forward_context.loop_enters:
1712 if forward_ctxt == grad_ctxt.grad_state.forward_context:1760 if op_input_forward_ctxt == grad_ctxt.grad_state.forward_context:1996 forward_ctxt = self.grad_state.forward_context