Searched refs:while_ctxt (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | control_flow_util.py | 191 while_ctxt = GetContainingWhileContext(op_ctxt)
194 if while_ctxt is None:
207 elif IsContainingContext(while_ctxt, input_while_ctxt):
211 elif (while_ctxt.grad_state and
212 IsContainingContext(while_ctxt.grad_state.forward_context,
221 elif (while_ctxt.grad_state and
222 while_ctxt.grad_state.forward_context is
229 input_while_ctxt.grad_state.forward_context is while_ctxt):
237 while_ctxt):
244 if while_ctxt:
[all …]
|
| D | control_flow_ops.py | 713 def GetMaxSizeFromNestedMaximumIterations(value, while_ctxt): argument
744 while while_ctxt not in (None, curr_ctxt):
745 max_iter = while_ctxt.maximum_iterations
750 "the tf.while_loop call ('%s')." % (value_name, while_ctxt.name))
772 (value_name, max_iter.name, while_ctxt.name, curr_ctxt_name,
778 while_ctxt = util.GetContainingWhileContext(
779 while_ctxt.outer_context, stop_ctxt=curr_ctxt)
1623 while_ctxt = self.GetWhileContext()
1626 if while_ctxt is None:
1632 if ctxt is not None and ctxt.GetWhileContext() == while_ctxt:
[all …]
|