Searched refs:while_loop_v2 (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | while_v2_test.py | 55 from tensorflow.python.ops.while_v2 import while_loop as while_loop_v2 unknown
85 ret = while_loop_v2(
95 ret = while_loop_v2(
124 ret = while_loop_v2(
134 ret = while_loop_v2(
153 while_loop_v2(lambda x: x < 10, Body, [x])
169 _, x = while_loop_v2(
203 _, z = while_loop_v2(
221 _, z, results = while_loop_v2(
238 _, b = while_loop_v2(
[all …]
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | xla_control_flow_ops_test.py | 103 return control_flow_ops.while_loop_v2(
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | remote_test.py | 533 return control_flow_ops.while_loop_v2(lambda _, d: d < 1, body, [i, 0])[0]
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | mirrored_strategy_test.py | 439 return control_flow_ops.while_loop_v2(lambda i: i < 2, body_fn, [0])
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | control_flow_ops.py | 2365 def while_loop_v2(cond, function
|