Home
last modified time | relevance | path

Searched refs:forward_context (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dcontrol_flow_util.py312 IsContainingContext(while_ctxt.grad_state.forward_context,
322 while_ctxt.grad_state.forward_context is
329 input_while_ctxt.grad_state.forward_context is while_ctxt):
335 input_ctxt.grad_state.forward_context.grad_state and
336 input_ctxt.grad_state.forward_context.grad_state.forward_context is
Dcontrol_flow_state.py156 outer_forward_ctxt = outer_grad_state.forward_context
214 def forward_context(self): member in _GradLoopState
344 value, self.forward_context)
351 enter_acc = self.forward_context.AddValue(acc)
354 swap_enabled = self.forward_context.swap_memory
356 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:
Dcontrol_flow_ops.py1712 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