OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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