| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | control_flow_ops_py_test.py | 47 from tensorflow.python.ops import control_flow_ops
133 _, r_s = control_flow_ops.while_loop(
146 v = control_flow_ops._Identity(v)
148 v2 = control_flow_ops.with_dependencies([op], v)
159 enter_v = control_flow_ops._Enter(v, "foo_1", is_constant=True)
163 v2 = control_flow_ops.with_dependencies([op], enter_v)
164 v3 = control_flow_ops.exit(v2)
174 v1 = control_flow_ops._SwitchRefOrTensor(v._ref(), p) # pylint: disable=protected-access
186 exit_op = control_flow_ops.exit(mul_op)
213 switch_op = control_flow_ops.switch(data, pred)
[all …]
|
| D | control_flow_util_test.py | 27 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(
|
| D | numerics_test.py | 28 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,
122 _ = control_flow_ops.while_loop(lambda _: predicate,
|
| D | critical_section_test.py | 28 from tensorflow.python.ops import control_flow_ops
74 return control_flow_ops.cond(
81 control_flow_ops.cond(array_ops.identity(outer_cond),
105 error = control_flow_ops.Assert((i % 2) == 1, ["Error"])
136 return control_flow_ops.no_op()
234 (i_n, j_n) = control_flow_ops.while_loop(
261 (i_n, j_n) = control_flow_ops.while_loop(
286 (i_n, j_n) = control_flow_ops.while_loop(
362 out = control_flow_ops.while_loop(
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | BUILD | 15 "control_flow_ops.py",
22 ":control_flow_ops",
28 "//tensorflow/python:control_flow_ops",
56 "//tensorflow/python:control_flow_ops",
74 name = "control_flow_ops",
75 srcs = ["control_flow_ops.py"],
81 "//tensorflow/python:control_flow_ops",
106 ":control_flow_ops",
126 ":control_flow_ops",
141 ":control_flow_ops",
[all …]
|
| D | gradients.py | 24 from tensorflow.python.ops.parallel_for import control_flow_ops
61 pfor_outputs = control_flow_ops.pfor(
64 pfor_outputs = control_flow_ops.for_loop(
129 pfor_output = control_flow_ops.pfor(loop_fn, output_row_size,
132 pfor_output = control_flow_ops.for_loop(
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | control_flow_ops_test.py | 36 from tensorflow.python.ops import control_flow_ops
72 control_flow_ops.group(a.op, b.op, c.op, name="root")
86 control_flow_ops.group(a.op, b.op, name="root")
103 control_flow_ops.group(a.op, b.op, c.op, d.op, name="root")
122 control_flow_ops.group([a.op, b.op], name="root")
133 control_flow_ops.group(1, 2)
142 control_flow_ops.with_dependencies(
153 const_with_dep = control_flow_ops.with_dependencies(
167 const_with_dep = control_flow_ops.with_dependencies(
189 _, switch_true = control_flow_ops.switch(data, less_op)
[all …]
|
| D | confusion_matrix.py | 26 from tensorflow.python.ops import control_flow_ops
80 predictions = control_flow_ops.cond(
86 labels = control_flow_ops.cond(
160 labels = control_flow_ops.with_dependencies(
164 predictions = control_flow_ops.with_dependencies(
174 labels = control_flow_ops.with_dependencies(
178 predictions = control_flow_ops.with_dependencies(
|
| D | standard_ops.py | 48 from tensorflow.python.ops.control_flow_ops import Assert
49 from tensorflow.python.ops.control_flow_ops import case
50 from tensorflow.python.ops.control_flow_ops import cond
51 from tensorflow.python.ops.control_flow_ops import group
52 from tensorflow.python.ops.control_flow_ops import no_op
53 from tensorflow.python.ops.control_flow_ops import tuple # pylint: disable=redefined-builtin
56 from tensorflow.python.ops.control_flow_ops import while_loop
|
| D | weights_broadcast_ops.py | 28 from tensorflow.python.ops import control_flow_ops
53 return control_flow_ops.cond(
97 return control_flow_ops.no_op(name="static_scalar_check_success")
117 return control_flow_ops.no_op(name="static_dims_check_success")
127 is_valid_shape = control_flow_ops.cond(
133 return control_flow_ops.Assert(is_valid_shape, data, name=scope)
|
| D | control_flow_grad.py | 26 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(
|
| D | check_ops.py | 32 from tensorflow.python.ops import control_flow_ops
512 return control_flow_ops.Assert(condition, data, summarize=summarize)
593 return control_flow_ops.Assert(condition, data, summarize=summarize)
717 return control_flow_ops.Assert(condition, data, summarize=summarize)
792 return control_flow_ops.Assert(condition, data, summarize=summarize)
868 return control_flow_ops.Assert(condition, data, summarize=summarize)
943 return control_flow_ops.Assert(condition, data, summarize=summarize)
1023 return control_flow_ops.Assert(condition, data, summarize=summarize)
1060 return control_flow_ops.no_op(name='static_checks_determined_all_ok')
1069 condition = control_flow_ops.with_dependencies([rank_check], condition)
[all …]
|
| /external/tensorflow/tensorflow/contrib/mixed_precision/python/ |
| D | loss_scale_manager.py | 25 from tensorflow.python.ops import control_flow_ops
151 return control_flow_ops.group(
166 new_loss_scale = control_flow_ops.cond(
172 return control_flow_ops.group(update_op, self._reset_stats())
174 return control_flow_ops.cond(
187 return control_flow_ops.group(update_op, self._reset_stats())
191 return control_flow_ops.group(
195 return control_flow_ops.cond(
199 return control_flow_ops.cond(finite_grads, update_if_finite_grads,
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/utils/ |
| D | losses.py | 24 from tensorflow.python.ops import control_flow_ops
32 return math_ops.square(loss), control_flow_ops.no_op()
50 return unweighted_loss * weights, control_flow_ops.no_op()
90 return unweighted_loss, control_flow_ops.no_op()
92 return unweighted_loss * weights, control_flow_ops.no_op()
152 return unweighted_loss, control_flow_ops.no_op()
154 return unweighted_loss * weights, control_flow_ops.no_op()
172 return unweighted_loss * weights, control_flow_ops.no_op()
260 return unweighted_loss * weights, control_flow_ops.no_op()
319 return unweighted_loss * weights, control_flow_ops.no_op()
|
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| D | sampling_ops.py | 25 from tensorflow.python.ops import control_flow_ops
116 control_flow_ops.with_dependencies([
218 assert_op = control_flow_ops.Assert(
226 init_probs = control_flow_ops.with_dependencies([assert_op], init_probs)
231 accept_probs = control_flow_ops.cond(
327 checked_probs = control_flow_ops.with_dependencies([
357 labels = control_flow_ops.with_dependencies([lbl_assert], labels)
359 control_flow_ops.with_dependencies([
366 labels = control_flow_ops.with_dependencies([
|
| D | bucket_ops.py | 36 from tensorflow.python.ops import control_flow_ops
235 control_flow_ops.cond(
237 functools.partial(enqueue_single, i), control_flow_ops.no_op)
240 return control_flow_ops.group(*enqueues, name="group_enqueues")
245 control_flow_ops.no_op)
279 close_op=control_flow_ops.group(
281 cancel_op=control_flow_ops.group(
|
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | multitask_optimizer_wrapper.py | 28 from tensorflow.python.ops import control_flow_ops
46 return control_flow_ops.no_op()
47 return control_flow_ops.cond(all_zeros, control_flow_ops.no_op, call_fn)
134 return control_flow_ops.cond(
|
| /external/tensorflow/tensorflow/contrib/integrate/python/ops/ |
| D | odes.py | 30 from tensorflow.python.ops import control_flow_ops
203 assert_op = control_flow_ops.Assert(
276 assert_increasing = control_flow_ops.Assert(
338 check_underflow = control_flow_ops.Assert(t0 + dt > t0,
340 check_max_num_steps = control_flow_ops.Assert(
342 check_numerics = control_flow_ops.Assert(
361 y_next = control_flow_ops.cond(accept_step, lambda: y1, lambda: y0)
362 f_next = control_flow_ops.cond(accept_step, lambda: f1, lambda: f0)
363 t_next = control_flow_ops.cond(accept_step, lambda: t0 + dt, lambda: t0)
364 interp_coeff = control_flow_ops.cond(
[all …]
|
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| D | gmm_ops.py | 30 from tensorflow.python.ops import control_flow_ops
202 initialized = control_flow_ops.with_dependencies(
208 init_means = control_flow_ops.with_dependencies(
211 self._init_ops.append(control_flow_ops.cond(initialized,
212 control_flow_ops.no_op,
216 init_covs = control_flow_ops.with_dependencies(
219 self._init_ops.append(control_flow_ops.cond(initialized,
220 control_flow_ops.no_op,
225 return control_flow_ops.group(*self._init_ops)
229 return control_flow_ops.group(*self._train_ops)
[all …]
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | auto_control_deps_test.py | 32 from tensorflow.python.ops import control_flow_ops
69 control_flow_ops.cond(p, true_fn, false_fn)
91 control_flow_ops.cond(p, true_fn, false_fn)
122 control_flow_ops.cond(q, inner_true_fn, inner_false_fn)
125 control_flow_ops.cond(p, true_fn, false_fn)
149 control_flow_ops.cond(p, true_fn, false_fn)
171 control_flow_ops.cond(p, true_fn, false_fn)
192 control_flow_ops.cond(p, true_fn, false_fn)
207 i, out = control_flow_ops.while_loop(c, b, (0, x))
309 control_flow_ops.Assert(x, ["Error"])
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | while_test.py | 31 from tensorflow.python.ops import control_flow_ops
138 xla_context = control_flow_ops.XLAControlFlowContext()
145 r = control_flow_ops.while_loop(
172 xla_context = control_flow_ops.XLAControlFlowContext()
180 r = control_flow_ops.while_loop(
191 return (i + 1, x + control_flow_ops.while_loop(
198 r = control_flow_ops.while_loop(
233 xla_context = control_flow_ops.XLAControlFlowContext()
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/ |
| D | weibull.py | 24 from tensorflow.python.ops import control_flow_ops
86 self._scale = control_flow_ops.with_dependencies([
91 self._concentration = control_flow_ops.with_dependencies([
141 return control_flow_ops.with_dependencies([is_valid], x)
151 return control_flow_ops.with_dependencies([is_positive, less_than_one], y)
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | losses_utils.py | 27 from tensorflow.python.ops import control_flow_ops
104 maybe_squeeze_dims = lambda: control_flow_ops.cond( # pylint: disable=g-long-lambda
106 y_true, y_pred = control_flow_ops.cond(
132 return control_flow_ops.cond(
138 return control_flow_ops.cond(
144 sample_weight = control_flow_ops.cond(
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_identity.py | 29 from tensorflow.python.ops import control_flow_ops
49 self._num_rows = control_flow_ops.with_dependencies([
307 return control_flow_ops.no_op("assert_non_singular")
310 return control_flow_ops.no_op("assert_positive_definite")
313 return control_flow_ops.no_op("assert_self_adjoint")
353 x = control_flow_ops.with_dependencies([aps], x)
402 self._num_rows = control_flow_ops.with_dependencies([
437 self._batch_shape_arg = control_flow_ops.with_dependencies([
663 x = control_flow_ops.with_dependencies([aps], x)
681 rhs = control_flow_ops.with_dependencies([aps], rhs)
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | main_op_impl.py | 22 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)
|