Home
last modified time | relevance | path

Searched refs:pfor_control_flow_ops (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/
H A Dcontrol_flow_ops_test.py74 from tensorflow.python.ops.parallel_for import control_flow_ops as pfor_control_flow_ops unknown
118 x = pfor_control_flow_ops.pfor(
126 pfor_control_flow_ops.pfor(lambda i: 1, 8, parallel_iterations=0)
128 pfor_control_flow_ops.for_loop(
133 pfor_control_flow_ops.pfor(lambda i: 1, 8, parallel_iterations=1)
140 result = pfor_control_flow_ops.vectorized_map(compute,
151 result = pfor_control_flow_ops.vectorized_map(compute, x)
159 result = pfor_control_flow_ops.vectorized_map(
175 a_plus_1 = pfor_control_flow_ops.vectorized_map(lambda a: a + 1, a)
179 a_plus_a = pfor_control_flow_ops.vectorized_map(sum, (a, a))
[all …]
H A Dxla_control_flow_ops_test.py31 from tensorflow.python.ops.parallel_for import control_flow_ops as pfor_control_flow_ops unknown
69 return pfor_control_flow_ops.vectorized_map(compute, x)
82 return pfor_control_flow_ops.vectorized_map(compute, x)
94 output = pfor_control_flow_ops.vectorized_map(compute, inp)
118 return pfor_control_flow_ops.pfor(loop_fn, 8)
158 return pfor_control_flow_ops.pfor(loop_fn, iters)
H A Dtest_util.py20 from tensorflow.python.ops.parallel_for import control_flow_ops as pfor_control_flow_ops unknown
55 t1 = pfor_control_flow_ops.pfor(
61 t2 = pfor_control_flow_ops.for_loop(loop_fn, loop_fn_dtypes, iters=iters,
H A Darray_test.py26 from tensorflow.python.ops.parallel_for import control_flow_ops as pfor_control_flow_ops unknown
87 r = pfor_control_flow_ops.vectorized_map(
236 pfor_control_flow_ops.pfor(loop_fn, 2, fallback_to_while_loop=False)
H A Dmath_test.py32 from tensorflow.python.ops.parallel_for import control_flow_ops as pfor_control_flow_ops unknown
242 r = pfor_control_flow_ops.vectorized_map(