| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | control_flow_ops_py_test.py | 137 _, r_s = control_flow_ops.while_loop(
420 r = control_flow_ops.while_loop(lambda i: i < 1000,
1424 return control_flow_ops.while_loop(
1444 def while_loop(): function
1448 self.assertEqual(self.evaluate(while_loop()), 2)
1484 r = control_flow_ops.while_loop(c, b, [n], parallel_iterations=20)
1498 result = control_flow_ops.while_loop(cond=lambda i: i < 2,
1515 result = control_flow_ops.while_loop(cond=lambda i: i < 5,
1534 r = control_flow_ops.while_loop(c, b, [i, x], parallel_iterations=5)
1561 r = control_flow_ops.while_loop(
[all …]
|
| D | control_flow_util_test.py | 81 def while_loop(x): function
91 return control_flow_ops.while_loop(c, b, [x])
96 math_ops.less(x, 1000), lambda: while_loop(x),
|
| D | while_v2_test.py | 49 from tensorflow.python.ops.while_v2 import while_loop as while_loop_v2
483 outputs = control_flow_ops.while_loop(Cond, Body, [x, tensor_list])
533 r = control_flow_ops.while_loop(
578 return control_flow_ops.while_loop(InnerCond, InnerBody,
581 outputs = control_flow_ops.while_loop(Cond, Body, [x, tensor_list])
625 r = control_flow_ops.while_loop(
637 return (i + 1, x + control_flow_ops.while_loop(
645 r = control_flow_ops.while_loop(
667 r = control_flow_ops.while_loop(
695 r = control_flow_ops.while_loop(
[all …]
|
| D | critical_section_test.py | 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(
368 out = control_flow_ops.while_loop(
|
| D | stack_ops_test.py | 82 r = control_flow_ops.while_loop(c, b, [n])
95 _, ry = control_flow_ops.while_loop(
234 r = control_flow_ops.while_loop(c, b, [n])
247 _, ry = control_flow_ops.while_loop(
|
| D | control_flow_util_v2_test.py | 48 return control_flow_ops.while_loop(lambda i: i < 4, body,
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | while_test.py | 56 loop_outputs = xla.while_loop([init_index], loop_cond, loop_body)
79 loop_outputs = xla.while_loop([init_index, init_sum], loop_cond,
105 loop_outputs = xla.while_loop([init_index, init_sum], loop_cond,
130 loop_outputs = xla.while_loop([init_index, 42], loop_cond, loop_body)
146 r = control_flow_ops.while_loop(
181 r = control_flow_ops.while_loop(
192 return (i + 1, x + control_flow_ops.while_loop(
199 r = control_flow_ops.while_loop(
|
| /external/swiftshader/third_party/llvm-7.0/llvm/test/tools/llvm-cov/ |
| D | deferred-region.cpp | 34 void while_loop() { in while_loop() function
84 while_loop(); in main()
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | training_loop.py | 30 def while_loop(condition, body, inputs=None, infeed_queue=None, name=None): function
177 return control_flow_ops.while_loop(
215 outputs = while_loop(
|
| /external/OpenCL-CTS/test_conformance/clcpp/attributes/ |
| D | test_ivdep.hpp | 33 while_loop, enumerator
79 else if (options.loop == loop_kind::while_loop) in generate_source()
98 else if (options.loop == loop_kind::while_loop) in generate_source()
145 else if (options.loop == loop_kind::while_loop) in generate_source()
165 else if (options.loop == loop_kind::while_loop) in generate_source()
365 options.loop = loop_kind::while_loop; in AUTO_TEST_CASE()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | control_flow_ops_test.py | 218 _, cost = control_flow_ops.while_loop(
240 _, cost = control_flow_ops.while_loop(
263 _, cost = control_flow_ops.while_loop(
308 _, outputs = control_flow_ops.while_loop(cond, body,
336 _, outputs = control_flow_ops.while_loop(cond, body,
488 control_flow_ops.while_loop(
946 iteration, matrix = control_flow_ops.while_loop(
1132 _, result = control_flow_ops.while_loop(cond, body, [0, 0.])
1292 r = control_flow_ops.while_loop(c, b, [i])
1301 r = control_flow_ops.while_loop(c, b, [i])
[all …]
|
| D | functional_ops.py | 147 _, r_a = control_flow_ops.while_loop(
341 _, r_a = control_flow_ops.while_loop(
655 _, _, r_a = control_flow_ops.while_loop(
|
| D | gradients_impl.py | 413 _, hessian = control_flow_ops.while_loop(
|
| /external/tensorflow/tensorflow/cc/ |
| D | BUILD | 39 "ops/while_loop.h",
69 ":while_loop",
108 ":while_loop",
301 name = "while_loop",
302 srcs = ["ops/while_loop.cc"],
303 hdrs = ["ops/while_loop.h"],
328 ":while_loop",
|
| /external/tensorflow/tensorflow/core/profiler/ |
| D | tfprof_log.proto | 106 // or in tf.while_loop.
109 // For cpu, vector size can be larger than 1 if in tf.while_loop.
113 // Normally, there will be multiple entries in while_loop.
|
| /external/tensorflow/tensorflow/python/autograph/g3doc/reference/ |
| D | limitations.md | 17 value (typically a `tf.Tensor`) modified by a loop. See `tf.while_loop`.
39 corresponding `tf.while_loop':
46 x = tf.while_loop(..., loop_vars=(x,)
59 For example, the symbol 'y' below is not wired through the `tf.while_loop`'s
69 x = tf.while_loop(..., loop_vars=(x,) # y does not need to be a loop variable
217 tf.while_loop(
226 for the `tf.while_loop`:
445 x = tf.while_loop(
467 In a `tf.while_loop` (and correspondingly, an AutoGraph `while` or `for` loop)
478 x = tf.while_loop(
|
| D | control_flow.md | 214 loops by converting them to `tf.while_loop`:
283 loops by converting them to a `tf.while_loop` which iterates over the first
297 of the `tf.while_loop`.
300 TensorFlow loops by converting them to a `tf.while_loop` which uses the
343 as TensorFlow loops by converting them to a combination of `tf.while_loop`,
|
| D | common_errors.md | 32 This usually indicates a loop that was intended to run as a `tf.while_loop`, but
104 within a `tf.cond`, `tf.while_loop`, or another `@tf.function` without using
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | control_flow_ops_test.py | 846 x = control_flow_ops.while_loop(lambda j: j < 4, lambda j: j + 1, [0])
857 return control_flow_ops.while_loop(lambda j: j < 4, lambda j: j + 1, [0])
866 return control_flow_ops.while_loop(lambda j: j < 4, lambda j: j + 1,
875 return control_flow_ops.while_loop(
885 return control_flow_ops.while_loop(lambda j, x: j < 4,
899 _, total = control_flow_ops.while_loop(
912 _, out = control_flow_ops.while_loop(
943 _, ta = control_flow_ops.while_loop(
961 _, s = control_flow_ops.while_loop(lambda j, _: j < i,
980 _, ta = control_flow_ops.while_loop(
[all …]
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | constant_folding_test.py | 56 _, y = control_flow_ops.while_loop(
|
| D | auto_mixed_precision_test.py | 146 ij = control_flow_ops.while_loop(c, b, init)
155 ijzw = control_flow_ops.while_loop(c, b, init)
197 r = control_flow_ops.while_loop(cond, body, init)
732 loss, _ = control_flow_ops.while_loop(
|
| /external/tensorflow/tensorflow/python/ops/signal/ |
| D | util_ops.py | 72 a, b = control_flow_ops.while_loop(cond, body, [a, b], back_prop=False)
|
| /external/tensorflow/tensorflow/examples/tf2_showcase/ |
| D | README.md | 21 - Don't use tf.cond or tf.while_loop; instead, make use of AutoGraph's
|
| /external/tensorflow/tensorflow/python/ops/linalg/sparse/ |
| D | conjugate_gradient.py | 134 _, state = control_flow_ops.while_loop(
|
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scaling_gradient_tape.py | 313 grads, _, _ = control_flow_ops.while_loop(
|