Home
last modified time | relevance | path

Searched refs:while_loop (Results 1 – 25 of 138) sorted by relevance

123456

/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/control_flow/
H A Dcontrol_flow_ops_py_test.py137 _, 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 …]
H A Dwhile_v2_test.py54 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 …]
H A Dcontrol_flow_util_test.py77 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),
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Dwhile_test.py52 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(
/aosp_15_r20/external/sdv/vsomeip/third_party/boost/phoenix/test/bll_compatibility/
Dcontrol_structures.cpp32 using boost::lambda::while_loop;
62 while_loop((_1 + _2) >= (_1 * _2), (++_1, ++_2, ++_3))(a, b, c); in simple_loops()
67 while_loop(_1++ < 10, ++boost::lambda::var(count))(i); in simple_loops()
81 while_loop(--_1 > _2)(a, b); in simple_loops()
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/
H A Dcontrol_flow_ops_test.py1391 _, 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 …]
H A Dindex.md5 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.
H A Dxla_control_flow_ops_test.py184 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(
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner_test.cc3922 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 …]
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dcontrol_flow_ops_test.py223 _, 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 …]
/aosp_15_r20/external/tensorflow/tensorflow/cc/
H A DBUILD44 "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",
/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/
H A Dtraining_loop.py30 def while_loop(condition: Callable[..., Any], function
177 return control_flow_ops.while_loop(
221 outputs = while_loop(
/aosp_15_r20/external/tensorflow/tensorflow/python/autograph/g3doc/reference/
H A Doperators.md132 `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`.
H A Dlimitations.md17 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(
H A D_control_flow_tutorial.ipynb75 …"`tf.cond` and `tf.while_loop` continue to work with `tf.function`, but code with control flow is …
377 …"If a loop is converted, it will be dynamically unrolled with `tf.while_loop`, or in the special c…
395 " print(\"{}({}) uses tf.while_loop.\".format(\n",
692 "As with `tf.cond`, `tf.while_loop` also comes with a number of subtleties.\n"
704 …"Since a loop can execute 0 times, all tensors used downstream of the while_loop must be initializ…
H A Dcontrol_flow.md214 loops by converting them to `tf.while_loop`:
284 loops by converting them to a `tf.while_loop` which iterates over the first
298 of the `tf.while_loop`.
301 TensorFlow loops by converting them to a `tf.while_loop` which uses the
344 as TensorFlow loops by converting them to a combination of `tf.while_loop`,
/aosp_15_r20/external/pytorch/torch/_higher_order_ops/
H A D__init__.py7 from torch._higher_order_ops.while_loop import while_loop
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/
H A Dcritical_section_test.py238 (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(
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/data_structures/
H A Dstack_ops_test.py78 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(
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_control_flow.py509 return torch._higher_order_ops.while_loop(cond_fn, body_fn, [ci, a, b])
523 i1, j1, x1, y1 = torch._higher_order_ops.while_loop(
529 return torch._higher_order_ops.while_loop(cond_fn, body_fn, (ci, cj, a, b))
547 return torch._higher_order_ops.while_loop(
562 _, f, g = torch._higher_order_ops.while_loop(cond_fn, body_fn, [c, d, e])
581 return torch._higher_order_ops.while_loop(cond_fn, body_fn, [c, a, b])
/aosp_15_r20/external/tensorflow/tensorflow/core/profiler/
H A Dtfprof_log.proto107 // 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.
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/lhlo_to_jitrt/
H A Dwhile_to_jitrt.mlir32 // CHECK: @while_loop(
36 func.func @while_loop(%arg0: memref<i32>, %arg1: memref<i1>) {
/aosp_15_r20/external/executorch/exir/
H A Dcontrol_flow.py129 def while_loop( function
169 while_loop,
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/
H A Dcpp_wrapper_cpu.py1921 def codegen_while_loop(self, while_loop): argument
1922 name = while_loop.get_name()
1924 buf.codegen_reference() for buf in while_loop.carried_inputs
1927 buf.codegen_reference() for buf in while_loop.additional_inputs
1935 for inp, out in zip(outer_carried_inputs, while_loop.outputs):
1968 self.writeline(EnterSubgraphLine(self, while_loop.cond_subgraph.graph))
1970 while_loop.cond_subgraph, cond_outer_inputs, cond_outer_outputs
1981 self.writeline(EnterSubgraphLine(self, while_loop.body_subgraph.graph))
1983 while_loop.body_subgraph, body_outer_inputs, body_outer_outputs
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dauto_control_deps_test.py312 return control_flow_ops.while_loop(
421 return control_flow_ops.while_loop(
502 return control_flow_ops.while_loop(
618 return control_flow_ops.while_loop(
809 i, out = control_flow_ops.while_loop(c, b, (0, x))

123456