| /external/tensorflow/tensorflow/python/kernel_tests/control_flow/ |
| D | control_flow_ops_py_test.py | 137 _, r_s = control_flow_ops.while_loop(
429 r = control_flow_ops.while_loop(lambda i: i < 1000,
1496 return control_flow_ops.while_loop(
1516 def while_loop(): function
1520 self.assertEqual(self.evaluate(while_loop()), 2)
1556 r = control_flow_ops.while_loop(c, b, [n], parallel_iterations=20)
1570 result = control_flow_ops.while_loop(cond=lambda i: i < 2,
1587 result = control_flow_ops.while_loop(cond=lambda i: i < 5,
1606 r = control_flow_ops.while_loop(c, b, [i, x], parallel_iterations=5)
1633 r = control_flow_ops.while_loop(
[all …]
|
| D | control_flow_util_test.py | 77 def while_loop(x): function
87 return control_flow_ops.while_loop(c, b, [x])
92 math_ops.less(x, 1000), lambda: while_loop(x),
|
| D | while_v2_test.py | 54 from tensorflow.python.ops.while_v2 import while_loop as while_loop_v2
287 _, z, results = control_flow_ops.while_loop(
675 outputs = control_flow_ops.while_loop(Cond, Body, [x, tensor_list])
725 r = control_flow_ops.while_loop(
770 return control_flow_ops.while_loop(InnerCond, InnerBody,
773 outputs = control_flow_ops.while_loop(Cond, Body, [x, tensor_list])
817 r = control_flow_ops.while_loop(
829 return (i + 1, x + control_flow_ops.while_loop(
837 r = control_flow_ops.while_loop(
859 r = control_flow_ops.while_loop(
[all …]
|
| D | control_flow_util_v2_test.py | 44 return control_flow_ops.while_loop(lambda i: i < 4, body,
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | while_test.py | 52 loop_outputs = xla.while_loop([init_index], loop_cond, loop_body)
75 loop_outputs = xla.while_loop([init_index, init_sum], loop_cond,
101 loop_outputs = xla.while_loop([init_index, init_sum], loop_cond,
126 loop_outputs = xla.while_loop([init_index, 42], loop_cond, loop_body)
142 r = control_flow_ops.while_loop(
177 r = control_flow_ops.while_loop(
188 return (i + 1, x + control_flow_ops.while_loop(
195 r = control_flow_ops.while_loop(
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | training_loop.py | 30 def while_loop(condition: Callable[..., Any], function
177 return control_flow_ops.while_loop(
221 outputs = while_loop(
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | xla_control_flow_ops_test.py | 184 return control_flow_ops.while_loop(
198 _, output = control_flow_ops.while_loop(
209 return control_flow_ops.while_loop(
235 return control_flow_ops.while_loop(
|
| D | index.md | 5 TensorFlow provides in-graph looping constructs like `tf.while_loop` which are
13 `tf.while_loop` like `tf.map_fn`, `tf.vectorized_map` is implemented by using
114 in these cases, adding a `tf.while_loop` in place of a real vectorization for
115 the op. This is safe but generally slower. `tf.while_loop` can run iterations in
144 with. One option is to use the fallback `tf.while_loop` converter for these ops,
156 structure). Stateless random ops [use the `tf.while_loop` fallback converter](https://github.com/te…
160 ## Vectorization of control flow (while_loop, cond) and variants
181 `tf.while_loop` and then request vectorization for it (although the fallback
188 `tf.while_loop`s the single vectorized loop terminates.
|
| D | control_flow_ops_test.py | 1391 _, out_handle = control_flow_ops.while_loop(
1403 _, out_handle = control_flow_ops.while_loop(
1416 _, handle = control_flow_ops.while_loop(
1420 _, out_handle = control_flow_ops.while_loop(
1433 _, out_handle = control_flow_ops.while_loop(
1653 x = control_flow_ops.while_loop(lambda j: j < 4, lambda j: j + 1, [0])
1664 return control_flow_ops.while_loop(lambda j: j < 4, lambda j: j + 1, [0])
1673 return control_flow_ops.while_loop(lambda j: j < 4, lambda j: j + 1,
1682 return control_flow_ops.while_loop(
1692 return control_flow_ops.while_loop(lambda j, x: j < 4, lambda j, x:
[all …]
|
| /external/tensorflow/tensorflow/cc/ |
| D | BUILD | 44 "ops/while_loop.h",
72 ":while_loop",
111 ":while_loop",
307 name = "while_loop",
308 srcs = ["ops/while_loop.cc"],
309 hdrs = ["ops/while_loop.h"],
334 ":while_loop",
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | critical_section_test.py | 238 (i_n, j_n) = control_flow_ops.while_loop(
265 (i_n, j_n) = control_flow_ops.while_loop(
290 (i_n, j_n) = control_flow_ops.while_loop(
365 out = control_flow_ops.while_loop(
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | control_flow_ops_test.py | 223 _, cost = control_flow_ops.while_loop(
245 _, cost = control_flow_ops.while_loop(
268 _, cost = control_flow_ops.while_loop(
313 _, outputs = control_flow_ops.while_loop(cond, body,
341 _, outputs = control_flow_ops.while_loop(cond, body,
491 control_flow_ops.while_loop(
989 iteration, matrix = control_flow_ops.while_loop(
1189 _, result = control_flow_ops.while_loop(cond, body, [0, 0.])
1458 r = control_flow_ops.while_loop(c, b, [i])
1467 r = control_flow_ops.while_loop(c, b, [i])
[all …]
|
| /external/tensorflow/tensorflow/python/autograph/g3doc/reference/ |
| D | operators.md | 132 `tf.while_loop` requires that all loop variables have supported types (e.g.
208 * `tf.Tensor` produced by `tf.range`: to `tf.while_loop`, removing the
210 * `tf.Tensor`, `tf.RagedTensor`: to `tf.while_loop`, checking the loop vars
211 for consistency. `opts` forwarded to `tf.while_loop`. Iterates over the
215 * `tf.data.Iterator`, `tf.distribute.Iterator`: to `tf.while_loop` called on
316 * `tf.Tensor`: to `tf.while_loop`.
|
| D | limitations.md | 17 value (typically a `tf.Tensor`) modified by a loop. See `tf.while_loop`.
175 corresponding `tf.while_loop':
182 x = tf.while_loop(..., loop_vars=(x,)
195 For example, the symbol 'y' below is not wired through the `tf.while_loop`'s
205 x = tf.while_loop(..., loop_vars=(x,) # y does not need to be a loop variable
422 tf.while_loop(
431 for the `tf.while_loop`:
650 x = tf.while_loop(
672 In a `tf.while_loop` (and correspondingly, an AutoGraph `while` or `for` loop)
683 x = tf.while_loop(
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/lhlo_to_jitrt/ |
| D | while_to_jitrt.mlir | 32 // CHECK: @while_loop(
36 func.func @while_loop(%arg0: memref<i32>, %arg1: memref<i1>) {
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_test.cc | 3922 const auto while_loop = root->operand(0); in TEST_F() local
3932 EXPECT_THAT(while_loop->while_body()->root_instruction(), in TEST_F()
3974 const auto while_loop = root->operand(0); in TEST_F() local
3984 EXPECT_THAT(while_loop->while_body()->root_instruction(), in TEST_F()
4359 const auto while_loop = root->operand(0); in TEST_F() local
4362 while_loop->while_condition()->root_instruction(), in TEST_F()
4379 EXPECT_THAT(while_loop->while_body()->root_instruction(), in TEST_F()
4386 while_loop->while_body()->root_instruction()->operand(2); in TEST_F()
4437 const auto while_loop = root->operand(1)->operand(0); in TEST_F() local
4440 while_loop->while_condition()->root_instruction(), in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/data_structures/ |
| D | stack_ops_test.py | 78 r = control_flow_ops.while_loop(c, b, [n])
91 _, ry = control_flow_ops.while_loop(
230 r = control_flow_ops.while_loop(c, b, [n])
243 _, ry = control_flow_ops.while_loop(
|
| /external/tensorflow/tensorflow/core/profiler/ |
| D | tfprof_log.proto | 107 // or in tf.while_loop.
110 // For cpu, vector size can be larger than 1 if in tf.while_loop.
114 // Normally, there will be multiple entries in while_loop.
|
| /external/skia/resources/sksl/runtime_errors/ |
| D | IllegalStatements.rts | 7 int while_loop(int x) { while (x < 1) { x++; } return x; }
|
| /external/skia/tests/sksl/runtime_errors/ |
| D | IllegalStatements.skvm | 10 int while_loop(int x) { while (x < 1) { x++; } return x; }
|
| /external/tensorflow/tensorflow/python/ops/signal/ |
| D | util_ops.py | 68 a, b = control_flow_ops.while_loop(cond, body, [a, b], back_prop=False)
|
| /external/tensorflow/tensorflow/lite/testing/op_tests/ |
| D | tensor_list_dynamic_shape.py | 54 _, tensor_list = tf.while_loop(condition, loop_body, init_state)
|
| D | while_loop.py | 73 counter, value, result_increment_value = tf.while_loop(
|
| /external/tensorflow/tensorflow/lite/experimental/mlir/testing/op_tests/ |
| D | tensor_list_dynamic_shape.py | 58 _, tensor_list = tf.while_loop(condition, loop_body, init_state)
|
| /external/tensorflow/tensorflow/core/tfrt/saved_model/tests/ |
| D | gen_while_v1.py | 46 r = control_flow_ops.while_loop(lambda a: a < ten,
|