Home
last modified time | relevance | path

Searched refs:GetWhileContext (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dcontrol_flow_state.py506 forward_ctxt = forward_ctxt.GetWhileContext()
508 forward_ctxt = util.GetWhileContext(op)
573 forward_ctxt = util.GetWhileContext(op)
579 outer_forward_ctxt = outer_forward_ctxt.GetWhileContext()
611 outer_forward_ctxt = outer_forward_ctxt.GetWhileContext()
666 forward_ctxt = util.GetWhileContext(op)
Dcontrol_flow_util.py366 def GetWhileContext(op): function
370 ctxt = ctxt.GetWhileContext()
Dcontrol_flow_ops.py768 def GetWhileContext(self): member in ControlFlowContext
771 return self._outer_context.GetWhileContext()
776 while_ctxt = self.GetWhileContext()
785 if ctxt is not None and ctxt.GetWhileContext() == while_ctxt:
892 if self.GetWhileContext():
893 return self.GetWhileContext().grad_state
898 if self.GetWhileContext():
899 self.GetWhileContext().back_prop
1678 def GetWhileContext(self): member in WhileContext
1705 grad_ctxt = grad_ctxt.GetWhileContext()
[all …]
/external/tensorflow/tensorflow/python/compiler/xla/
Dxla.py306 if self.GetWhileContext():
307 return self.GetWhileContext().back_prop
/external/tensorflow/tensorflow/python/tpu/
Dtpu.py684 if self.GetWhileContext():
685 return self.GetWhileContext().back_prop