Home
last modified time | relevance | path

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

12345

/external/tensorflow/tensorflow/python/kernel_tests/
Dcontrol_flow_ops_py_test.py133 _, r_s = control_flow_ops.while_loop(
405 r = control_flow_ops.while_loop(lambda i: i < 1000,
1226 return control_flow_ops.while_loop(
1248 def while_loop(): function
1252 self.assertEqual(self.evaluate(while_loop()), 2)
1293 r = control_flow_ops.while_loop(c, b, [n], parallel_iterations=20)
1307 result = control_flow_ops.while_loop(cond=lambda i: i < 2,
1324 result = control_flow_ops.while_loop(cond=lambda i: i < 5,
1343 r = control_flow_ops.while_loop(c, b, [i, x], parallel_iterations=5)
1370 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.py234 (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(
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(
Dwhile_v2_test.py41 from tensorflow.python.ops.control_flow_ops import while_loop as while_loop_v1
42 from tensorflow.python.ops.while_v2 import while_loop as while_loop_v2
374 output = while_v2.while_loop(
461 output = while_v2.while_loop(lambda x: x < 10.0,
Dcontrol_flow_util_v2_test.py48 return control_flow_ops.while_loop(lambda i: i < 4, body,
/external/tensorflow/tensorflow/compiler/tests/
Dwhile_test.py55 loop_outputs = xla.while_loop([init_index], loop_cond, loop_body)
78 loop_outputs = xla.while_loop([init_index, init_sum], loop_cond,
104 loop_outputs = xla.while_loop([init_index, init_sum], loop_cond,
129 loop_outputs = xla.while_loop([init_index, 42], loop_cond, loop_body)
145 r = control_flow_ops.while_loop(
180 r = control_flow_ops.while_loop(
191 return (i + 1, x + control_flow_ops.while_loop(
198 r = control_flow_ops.while_loop(
/external/swiftshader/third_party/llvm-7.0/llvm/test/tools/llvm-cov/
Ddeferred-region.cpp34 void while_loop() { in while_loop() function
84 while_loop(); in main()
/external/tensorflow/tensorflow/python/tpu/
Dtraining_loop.py30 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/tensorflow/tensorflow/cc/
DBUILD19 "ops/while_loop.h",
49 ":while_loop",
88 ":while_loop",
276 name = "while_loop",
277 srcs = ["ops/while_loop.cc"],
278 hdrs = ["ops/while_loop.h"],
303 ":while_loop",
/external/tensorflow/tensorflow/contrib/opt/python/training/
Dmatrix_functions.py68 _, _, prev_mat_y, _, _, _, _ = control_flow_ops.while_loop(
153 _, _, mat_h = control_flow_ops.while_loop(
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dcontrol_flow_ops_test.py678 x = control_flow_ops.while_loop(lambda j: j < 4, lambda j: j + 1, [0])
689 return control_flow_ops.while_loop(lambda j: j < 4, lambda j: j + 1, [0])
698 return control_flow_ops.while_loop(lambda j: j < 4, lambda j: j + 1,
707 return control_flow_ops.while_loop(
717 return control_flow_ops.while_loop(lambda j, x: j < 4,
731 _, total = control_flow_ops.while_loop(
744 _, out = control_flow_ops.while_loop(
775 _, ta = control_flow_ops.while_loop(
793 _, s = control_flow_ops.while_loop(lambda j, _: j < i,
812 _, ta = control_flow_ops.while_loop(
[all …]
/external/tensorflow/tensorflow/core/profiler/
Dtfprof_log.proto106 // 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/ops/
Dcontrol_flow_ops_test.py207 _, cost = control_flow_ops.while_loop(
229 _, cost = control_flow_ops.while_loop(
252 _, cost = control_flow_ops.while_loop(
297 _, outputs = control_flow_ops.while_loop(cond, body,
325 _, outputs = control_flow_ops.while_loop(cond, body,
443 control_flow_ops.while_loop(
901 iteration, matrix = control_flow_ops.while_loop(
994 r = control_flow_ops.while_loop(c, b, [i])
1003 r = control_flow_ops.while_loop(c, b, [i])
1017 r = control_flow_ops.while_loop(c, b, [i, []])
[all …]
Dfunctional_ops.py140 _, r_a = control_flow_ops.while_loop(
250 _, r_a = control_flow_ops.while_loop(
471 _, _, r_a = control_flow_ops.while_loop(
/external/tensorflow/tensorflow/contrib/solvers/python/ops/
Dlanczos.py130 _, _, v = control_flow_ops.while_loop(lambda j, basis, v: j < i,
186 _, ls = control_flow_ops.while_loop(stopping_criterion, lanczos_bidiag_step,
Dleast_squares.py106 _, state = control_flow_ops.while_loop(stopping_criterion, cgls_step,
/external/tensorflow/tensorflow/contrib/integrate/python/ops/
Dodes.py380 rk_state, history, _ = control_flow_ops.while_loop(
402 solution, history, _, _ = control_flow_ops.while_loop(
581 solution_array, _, _, _ = control_flow_ops.while_loop(
607 y, t, _, _ = control_flow_ops.while_loop(
/external/tensorflow/tensorflow/contrib/losses/python/metric_learning/
Dmetric_loss_ops.py717 _, nmi_scores = control_flow_ops.while_loop(
766 _, chosen_ids = control_flow_ops.while_loop(
819 _, scores_margin = control_flow_ops.while_loop(func_cond, func_body,
875 _, chosen_ids = control_flow_ops.while_loop(
948 _, gt_cluster_score = control_flow_ops.while_loop(
/external/tensorflow/tensorflow/python/autograph/
DLIMITATIONS.md3 …`while` and AutoGraph automatically converts it into the equivalent `tf.cond`, and `tf.while_loop`.
14 … known lengths, as well as for loops over TF datasets. Converts to `tf.while_loop`, with an additi…
15 While statement | Yes | | Converts to `tf.while_loop`. Crea…
/external/tensorflow/tensorflow/python/grappler/
Dconstant_folding_test.py56 _, y = control_flow_ops.while_loop(
/external/tensorflow/tensorflow/python/ops/signal/
Dutil_ops.py66 a, b = control_flow_ops.while_loop(cond, body, [a, b], back_prop=False)
/external/tensorflow/tensorflow/examples/tf2_showcase/
DREADME.md21 - Don't use tf.cond or tf.while_loop; instead, make use of AutoGraph's
/external/tensorflow/tensorflow/contrib/cmake/
Dtf_cc_ops.cmake112 "${tensorflow_source_dir}/tensorflow/cc/ops/while_loop.h"
113 "${tensorflow_source_dir}/tensorflow/cc/ops/while_loop.cc"
/external/tensorflow/tensorflow/contrib/constrained_optimization/python/
Dswap_regret_optimizer.py118 iteration, eigenvector, old_eigenvector = control_flow_ops.while_loop(
190 iteration, matrix, inactive, old_inactive = control_flow_ops.while_loop(

12345