Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/kernel_tests/
Dcontrol_flow_util_test.py70 self.assertTrue(control_flow_util.IsLoopExit(exit_op))
73 self.assertTrue(control_flow_util.IsLoopExit(ref_exit))
75 self.assertFalse(control_flow_util.IsLoopExit(test_ops.int_output().op))
/external/tensorflow/tensorflow/python/ops/
Dcontrol_flow_util.py94 def IsLoopExit(op): function
173 if ctxt is not None and IsLoopExit(op):
Dcontrol_flow_ops.py1048 if before and util.IsLoopExit(op):
1313 if util.IsLoopExit(op):
1768 if self._outer_context or not util.IsLoopExit(op):
2417 if util.IsLoopExit(val.op):
2512 if self._outer_context or not util.IsLoopExit(op):
Dgradients_util.py802 if control_flow_util.IsLoopExit(x.op):
/external/tensorflow/tensorflow/python/tpu/
Dtensor_tracer.py469 control_flow_util.IsLoopExit(op) or
1406 if control_flow_util.IsLoopExit(op):