Home
last modified time | relevance | path

Searched refs:while_loop_v2 (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/control_flow/
H A Dwhile_v2_test.py54 from tensorflow.python.ops.while_v2 import while_loop as while_loop_v2 unknown
84 ret = while_loop_v2(
94 ret = while_loop_v2(
123 ret = while_loop_v2(
133 ret = while_loop_v2(
152 while_loop_v2(lambda x: x < 10, Body, [x])
168 _, x = while_loop_v2(
187 x, = while_loop_v2(lambda i: True, Body, [0], maximum_iterations=1)
218 _, z = while_loop_v2(
236 _, z, results = while_loop_v2(
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Ddef_function_xla_jit_test.py375 return control_flow_ops.while_loop_v2(
394 y, shp = control_flow_ops.while_loop_v2(
397 return control_flow_ops.while_loop_v2(
401 return control_flow_ops.while_loop_v2(
419 y, shp = control_flow_ops.while_loop_v2(
421 return control_flow_ops.while_loop_v2(
425 x = control_flow_ops.while_loop_v2(
429 w = control_flow_ops.while_loop_v2(
H A Dremote_test.py573 return control_flow_ops.while_loop_v2(lambda _, d: d < 1, body, [i, 0])[0]
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/
H A Dxla_control_flow_ops_test.py99 return control_flow_ops.while_loop_v2(
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dextension_type_test.py625 _, y = control_flow_ops.while_loop_v2(cond, body, [0, x])
657 control_flow_ops.while_loop_v2(cond, body, [0, x])
667 _, y = control_flow_ops.while_loop_v2(cond, body, [0, x])
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dmirrored_strategy_test.py463 return control_flow_ops.while_loop_v2(lambda i: i < 2, body_fn, [0])
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dcontrol_flow_ops.py2347 def while_loop_v2(cond, function