HomeSort by relevance Sort by last modified time
    Searched refs:control_flow_util (Results 1 - 7 of 7) sorted by null

  /external/tensorflow/tensorflow/python/kernel_tests/
control_flow_util_test.py 16 """Tests for tensorflow.python.ops.control_flow_util."""
24 from tensorflow.python.ops import control_flow_util
34 self.assertTrue(control_flow_util.IsSwitch(switch))
39 self.assertTrue(control_flow_util.IsSwitch(ref_switch))
41 self.assertFalse(control_flow_util.IsSwitch(test_ops.int_output().op))
45 self.assertTrue(control_flow_util.IsLoopEnter(enter))
46 self.assertFalse(control_flow_util.IsLoopConstantEnter(enter))
50 self.assertTrue(control_flow_util.IsLoopEnter(ref_enter))
51 self.assertFalse(control_flow_util.IsLoopConstantEnter(ref_enter))
55 self.assertTrue(control_flow_util.IsLoopEnter(const_enter)
    [all...]
  /external/tensorflow/tensorflow/contrib/tpu/python/tpu/
tpu_test.py 29 from tensorflow.python.ops import control_flow_util
38 """Test that control_flow_util can check that we're in a TPU context."""
44 self.assertFalse(control_flow_util.IsInXLAContext(z1.op))
45 self.assertTrue(control_flow_util.IsInXLAContext(z2.op))
  /external/tensorflow/tensorflow/python/ops/
control_flow_grad.py 26 from tensorflow.python.ops import control_flow_util
96 op_ctxt = control_flow_util.GetOutputContext(input_op)
gradients_impl.py 41 from tensorflow.python.ops import control_flow_util
674 pending_count[x.op._id] > 0 and control_flow_util.IsLoopSwitch(x.op))
677 if control_flow_util.IsLoopExit(x.op):
717 assert control_flow_util.IsLoopSwitch(op)
    [all...]
array_grad.py 30 from tensorflow.python.ops import control_flow_util
126 grad_context = control_flow_util.GetOutputContext(grad.op)
127 dim_context = control_flow_util.GetOutputContext(concat_dim.op)
control_flow_ops.py 71 from tensorflow.python.ops import control_flow_util as util
    [all...]
  /external/tensorflow/tensorflow/python/framework/
ops.py 52 from tensorflow.python.ops import control_flow_util
    [all...]

Completed in 549 milliseconds