Home
last modified time | relevance | path

Searched refs:control_flow_ops (Results 1 – 25 of 336) sorted by relevance

12345678910>>...14

/external/tensorflow/tensorflow/compiler/tests/
Dcond_test.py31 from tensorflow.python.ops import control_flow_ops
44 xla_context = control_flow_ops.XLAControlFlowContext()
50 output = control_flow_ops.cond(
64 xla_context = control_flow_ops.XLAControlFlowContext()
70 output = control_flow_ops.cond(
85 xla_context = control_flow_ops.XLAControlFlowContext()
90 output = control_flow_ops.cond(
103 xla_context = control_flow_ops.XLAControlFlowContext()
118 output = control_flow_ops.cond(
132 xla_context = control_flow_ops.XLAControlFlowContext()
[all …]
Dwhile_test.py32 from tensorflow.python.ops import control_flow_ops
139 xla_context = control_flow_ops.XLAControlFlowContext()
146 r = control_flow_ops.while_loop(
173 xla_context = control_flow_ops.XLAControlFlowContext()
181 r = control_flow_ops.while_loop(
192 return (i + 1, x + control_flow_ops.while_loop(
199 r = control_flow_ops.while_loop(
234 xla_context = control_flow_ops.XLAControlFlowContext()
248 xla_context = control_flow_ops.XLAControlFlowContext()
/external/tensorflow/tensorflow/python/kernel_tests/
Dcontrol_flow_ops_py_test.py53 from tensorflow.python.ops import control_flow_ops
140 _, r_s = control_flow_ops.while_loop(
172 v = control_flow_ops._Identity(v)
174 v2 = control_flow_ops.with_dependencies([op], v)
185 enter_v = control_flow_ops._Enter(v, "foo_1", is_constant=True)
189 v2 = control_flow_ops.with_dependencies([op], enter_v)
190 v3 = control_flow_ops.exit(v2)
200 v1 = control_flow_ops._SwitchRefOrTensor(v._ref(), p) # pylint: disable=protected-access
212 exit_op = control_flow_ops.exit(mul_op)
239 switch_op = control_flow_ops.switch(data, pred)
[all …]
Dcontrol_flow_util_test.py27 from tensorflow.python.ops import control_flow_ops
38 switch_false, _ = control_flow_ops.switch(1, True)
42 ref_switch_false, _ = control_flow_ops.ref_switch(test_ops.ref_output(),
69 exit_op = control_flow_ops.exit(1).op
72 ref_exit = control_flow_ops.exit(test_ops.ref_output()).op
85 return control_flow_ops.cond(
91 return control_flow_ops.while_loop(c, b, [x])
95 control_flow_ops.cond(
Dnumerics_test.py28 from tensorflow.python.ops import control_flow_ops
76 a = control_flow_ops.with_dependencies([check], a)
86 a = control_flow_ops.with_dependencies([check], a)
96 a = control_flow_ops.with_dependencies([check], a)
110 _ = control_flow_ops.cond(predicate,
121 _ = control_flow_ops.while_loop(lambda _: predicate,
Dcritical_section_test.py32 from tensorflow.python.ops import control_flow_ops
85 return control_flow_ops.cond(
92 control_flow_ops.cond(array_ops.identity(outer_cond),
116 error = control_flow_ops.Assert((i % 2) == 1, ["Error"])
147 return control_flow_ops.no_op()
242 (i_n, j_n) = control_flow_ops.while_loop(
269 (i_n, j_n) = control_flow_ops.while_loop(
294 (i_n, j_n) = control_flow_ops.while_loop(
369 out = control_flow_ops.while_loop(
Dcontrol_flow_util_v2_test.py25 from tensorflow.python.ops import control_flow_ops
46 return control_flow_ops.cond(constant_op.constant(True),
48 return control_flow_ops.while_loop(lambda i: i < 4, body,
/external/tensorflow/tensorflow/python/ops/parallel_for/
DBUILD14 "control_flow_ops.py",
21 ":control_flow_ops",
29 "//tensorflow/python:control_flow_ops",
59 "//tensorflow/python:control_flow_ops",
77 name = "control_flow_ops",
78 srcs = ["control_flow_ops.py"],
84 "//tensorflow/python:control_flow_ops",
119 ":control_flow_ops",
146 ":control_flow_ops",
165 ":control_flow_ops",
[all …]
Dgradients.py24 from tensorflow.python.ops.parallel_for import control_flow_ops
63 pfor_outputs = control_flow_ops.pfor(
66 pfor_outputs = control_flow_ops.for_loop(
134 pfor_output = control_flow_ops.pfor(loop_fn, output_row_size,
137 pfor_output = control_flow_ops.for_loop(
Dxla_control_flow_ops_test.py30 from tensorflow.python.ops import control_flow_ops
35 from tensorflow.python.ops.parallel_for import control_flow_ops as pfor_control_flow_ops
103 return control_flow_ops.while_loop_v2(
188 return control_flow_ops.while_loop(
199 _, output = control_flow_ops.while_loop(
210 return control_flow_ops.while_loop(
236 return control_flow_ops.while_loop(
/external/tensorflow/tensorflow/python/ops/
Dcontrol_flow_ops_test.py43 from tensorflow.python.ops import control_flow_ops
84 control_flow_ops.group(a.op, b.op, c.op, name="root")
99 control_flow_ops.group(a.op, b.op, name="root")
117 control_flow_ops.group(a.op, b.op, c.op, d.op, name="root")
137 control_flow_ops.group([a.op, b.op], name="root")
149 control_flow_ops.group(1, 2)
158 control_flow_ops.with_dependencies(
169 const_with_dep = control_flow_ops.with_dependencies(
182 const_with_dep = control_flow_ops.with_dependencies(
203 _, switch_true = control_flow_ops.switch(data, less_op)
[all …]
Dstandard_ops.py51 from tensorflow.python.ops.control_flow_ops import Assert
52 from tensorflow.python.ops.control_flow_ops import case
53 from tensorflow.python.ops.control_flow_ops import cond
54 from tensorflow.python.ops.control_flow_ops import group
55 from tensorflow.python.ops.control_flow_ops import no_op
56 from tensorflow.python.ops.control_flow_ops import tuple # pylint: disable=redefined-builtin
59 from tensorflow.python.ops.control_flow_ops import while_loop
111 from tensorflow.python.ops.parallel_for.control_flow_ops import vectorized_map
Dweights_broadcast_ops.py28 from tensorflow.python.ops import control_flow_ops
54 return control_flow_ops.cond(
98 return control_flow_ops.no_op(name="static_scalar_check_success")
118 return control_flow_ops.no_op(name="static_dims_check_success")
128 is_valid_shape = control_flow_ops.cond(
134 return control_flow_ops.Assert(is_valid_shape, data, name=scope)
173 if control_flow_ops.get_enclosing_xla_context() is not None:
Dconfusion_matrix.py25 from tensorflow.python.ops import control_flow_ops
81 predictions = control_flow_ops.cond(
87 labels = control_flow_ops.cond(
162 labels = control_flow_ops.with_dependencies(
166 predictions = control_flow_ops.with_dependencies(
176 labels = control_flow_ops.with_dependencies(
180 predictions = control_flow_ops.with_dependencies(
Dcontrol_flow_grad.py26 from tensorflow.python.ops import control_flow_ops
31 from tensorflow.python.ops.control_flow_ops import *
56 control_flow_ops._AddNextAndBackEdge(merge_grad, grad[1],
106 return control_flow_ops._SwitchRefOrTensor(grad, grad_ctxt.pivot)
130 return control_flow_ops._SwitchRefOrTensor(grad, pred, name="cond_grad")
136 return [control_flow_ops._SwitchRefOrTensor(grad, cond[i])[1]
175 result = control_flow_ops._Enter(
/external/tensorflow/tensorflow/python/framework/
Dauto_control_deps_test.py32 from tensorflow.python.ops import control_flow_ops
223 return control_flow_ops.while_loop(
238 return control_flow_ops.cond(
253 return control_flow_ops.cond(
268 return control_flow_ops.switch_case(
283 return control_flow_ops.switch_case(
332 return control_flow_ops.while_loop(
358 return control_flow_ops.cond(
413 return control_flow_ops.while_loop(
429 return control_flow_ops.cond(
[all …]
Dsmart_cond.py24 from tensorflow.python.ops import control_flow_ops
58 return control_flow_ops.cond(pred, true_fn=true_fn, false_fn=false_fn,
117 return control_flow_ops._case_helper( # pylint: disable=protected-access
/external/tensorflow/tensorflow/python/keras/utils/
Dlosses_utils.py26 from tensorflow.python.ops import control_flow_ops
133 predictions = control_flow_ops.cond(
139 labels = control_flow_ops.cond(
191 maybe_squeeze_dims = lambda: control_flow_ops.cond( # pylint: disable=g-long-lambda
193 y_true, y_pred = control_flow_ops.cond(
219 return control_flow_ops.cond(
223 return control_flow_ops.cond(
229 sample_weight = control_flow_ops.cond(
/external/tensorflow/tensorflow/python/ops/losses/
Dutil.py28 from tensorflow.python.ops import control_flow_ops
79 maybe_squeeze_dims = lambda: control_flow_ops.cond( # pylint: disable=g-long-lambda
81 y_true, y_pred = control_flow_ops.cond(
107 return control_flow_ops.cond(
111 return control_flow_ops.cond(
117 sample_weight = control_flow_ops.cond(
/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scale.py29 from tensorflow.python.ops import control_flow_ops
272 return control_flow_ops.no_op(), True
308 return control_flow_ops.cond(
310 control_flow_ops.no_op)
425 return control_flow_ops.group(
429 return control_flow_ops.cond(
439 return control_flow_ops.group(
443 update_op = control_flow_ops.cond(is_finite, update_if_finite_grads,
/external/tensorflow/tensorflow/python/saved_model/
Dmain_op_impl.py22 from tensorflow.python.ops import control_flow_ops
46 return control_flow_ops.group(init, init_local, init_tables)
71 main_op_with_restore = control_flow_ops.group(restore_op_name)
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/
Dcontrol_flow_upgrade_legacy_v1.py25 from tensorflow.python.ops import control_flow_ops
40 switch_op = control_flow_ops.switch(data, less_op)
41 merge_op = control_flow_ops.merge(switch_op)[0]
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_squeeze_op.py25 from tensorflow.python.ops import control_flow_ops
84 control_flow_ops.Assert(
90 control_flow_ops.Assert(
97 squeezed_rt = control_flow_ops.with_dependencies(assertion_list,
/external/tensorflow/tensorflow/python/ops/linalg/
DBUILD13 "//tensorflow/python:control_flow_ops",
33 "//tensorflow/python:control_flow_ops",
47 "//tensorflow/python:control_flow_ops",
/external/tensorflow/tensorflow/python/autograph/operators/
Dlogical.py22 from tensorflow.python.ops import control_flow_ops
54 return control_flow_ops.cond(cond, b, lambda: cond)
73 return control_flow_ops.cond(cond, lambda: cond, b)

12345678910>>...14