Home
last modified time | relevance | path

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

12345

/external/tensorflow/tensorflow/python/kernel_tests/
Dcontrol_flow_ops_py_test.py140 _, r_s = control_flow_ops.while_loop(
431 r = control_flow_ops.while_loop(lambda i: i < 1000,
1481 return control_flow_ops.while_loop(
1501 def while_loop(): function
1505 self.assertEqual(self.evaluate(while_loop()), 2)
1541 r = control_flow_ops.while_loop(c, b, [n], parallel_iterations=20)
1555 result = control_flow_ops.while_loop(cond=lambda i: i < 2,
1572 result = control_flow_ops.while_loop(cond=lambda i: i < 5,
1591 r = control_flow_ops.while_loop(c, b, [i, x], parallel_iterations=5)
1618 r = control_flow_ops.while_loop(
[all …]
Dcontrol_flow_util_test.py81 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),
Dcritical_section_test.py242 (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(
Dwhile_v2_test.py55 from tensorflow.python.ops.while_v2 import while_loop as while_loop_v2
272 _, z, results = control_flow_ops.while_loop(
660 outputs = control_flow_ops.while_loop(Cond, Body, [x, tensor_list])
710 r = control_flow_ops.while_loop(
755 return control_flow_ops.while_loop(InnerCond, InnerBody,
758 outputs = control_flow_ops.while_loop(Cond, Body, [x, tensor_list])
802 r = control_flow_ops.while_loop(
814 return (i + 1, x + control_flow_ops.while_loop(
822 r = control_flow_ops.while_loop(
844 r = control_flow_ops.while_loop(
[all …]
Dstack_ops_test.py82 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(
Dcontrol_flow_util_v2_test.py48 return control_flow_ops.while_loop(lambda i: i < 4, body,
/external/tensorflow/tensorflow/compiler/tests/
Dwhile_test.py56 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/llvm-project/llvm/test/tools/llvm-cov/
Ddeferred-region.cpp34 void while_loop() { in while_loop() function
84 while_loop(); in main()
/external/llvm-project/clang/test/CoverageMapping/
Ddeferred-region.cpp136 void while_loop() { in while_loop() function
212 while_loop(); in main()
/external/tensorflow/tensorflow/python/tpu/
Dtraining_loop.py34 def while_loop(condition: Callable[..., Any], function
185 return control_flow_ops.while_loop(
229 outputs = while_loop(
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dxla_control_flow_ops_test.py188 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(
Dcontrol_flow_ops_test.py1160 _, out_handle = control_flow_ops.while_loop(
1293 x = control_flow_ops.while_loop(lambda j: j < 4, lambda j: j + 1, [0])
1304 return control_flow_ops.while_loop(lambda j: j < 4, lambda j: j + 1, [0])
1313 return control_flow_ops.while_loop(lambda j: j < 4, lambda j: j + 1,
1322 return control_flow_ops.while_loop(
1332 return control_flow_ops.while_loop(lambda j, x: j < 4, lambda j, x:
1346 _, total = control_flow_ops.while_loop(
1359 _, out = control_flow_ops.while_loop(
1390 _, ta = control_flow_ops.while_loop(
1408 _, s = control_flow_ops.while_loop(lambda j, _: j < i, body, (0, 0))
[all …]
/external/tensorflow/tensorflow/cc/
DBUILD45 "ops/while_loop.h",
75 ":while_loop",
114 ":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/ops/
Dcontrol_flow_ops_test.py221 _, cost = control_flow_ops.while_loop(
243 _, cost = control_flow_ops.while_loop(
266 _, cost = control_flow_ops.while_loop(
311 _, outputs = control_flow_ops.while_loop(cond, body,
339 _, outputs = control_flow_ops.while_loop(cond, body,
489 control_flow_ops.while_loop(
987 iteration, matrix = control_flow_ops.while_loop(
1186 _, result = control_flow_ops.while_loop(cond, body, [0, 0.])
1455 r = control_flow_ops.while_loop(c, b, [i])
1464 r = control_flow_ops.while_loop(c, b, [i])
[all …]
/external/skia/resources/sksl/runtime_errors/
DIllegalStatements.rts7 int while_loop(int x) { while (x < 1) { x++; } return x; }
/external/tensorflow/tensorflow/core/profiler/
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.
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Dknown_issues.md25 TF while [loops](https://www.tensorflow.org/api_docs/python/tf/while_loop)
26 created using `tf.while_loop` support backpropagation by accumulating all
/external/tensorflow/tensorflow/python/ops/signal/
Dutil_ops.py72 a, b = control_flow_ops.while_loop(cond, body, [a, b], back_prop=False)
/external/tensorflow/tensorflow/python/autograph/g3doc/reference/
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(
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…
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`,
/external/llvm-project/clang/tools/clang-fuzzer/
Dcxx_proto.proto80 While while_loop = 3; field
/external/tensorflow/tensorflow/python/framework/
Dauto_control_deps_test.py223 return control_flow_ops.while_loop(
332 return control_flow_ops.while_loop(
413 return control_flow_ops.while_loop(
529 return control_flow_ops.while_loop(
720 i, out = control_flow_ops.while_loop(c, b, (0, x))
/external/tensorflow/tensorflow/c/experimental/saved_model/internal/testdata/
Dgen_saved_models.py86 acc, _ = control_flow_ops.while_loop(
/external/tensorflow/tensorflow/python/grappler/
Dconstant_folding_test.py61 _, y = control_flow_ops.while_loop(

12345