Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/ops/parallel_for/
Dcontrol_flow_ops_test.py72 from tensorflow.python.ops.parallel_for import control_flow_ops as pfor_control_flow_ops unknown
116 x = pfor_control_flow_ops.pfor(
124 pfor_control_flow_ops.pfor(lambda i: 1, 8, parallel_iterations=0)
126 pfor_control_flow_ops.for_loop(
131 pfor_control_flow_ops.pfor(lambda i: 1, 8, parallel_iterations=1)
138 result = pfor_control_flow_ops.vectorized_map(compute,
149 result = pfor_control_flow_ops.vectorized_map(compute, x)
162 a_plus_1 = pfor_control_flow_ops.vectorized_map(lambda a: a + 1, a)
166 a_plus_a = pfor_control_flow_ops.vectorized_map(sum, (a, a))
170 a_plus_b = pfor_control_flow_ops.vectorized_map(sum, (a, b))
[all …]
Dtest_util.py24 from tensorflow.python.ops.parallel_for import control_flow_ops as pfor_control_flow_ops unknown
59 t1 = pfor_control_flow_ops.pfor(
65 t2 = pfor_control_flow_ops.for_loop(loop_fn, loop_fn_dtypes, iters=iters,
Dxla_control_flow_ops_test.py35 from tensorflow.python.ops.parallel_for import control_flow_ops as pfor_control_flow_ops unknown
73 return pfor_control_flow_ops.vectorized_map(compute, x)
86 return pfor_control_flow_ops.vectorized_map(compute, x)
98 output = pfor_control_flow_ops.vectorized_map(compute, inp)
122 return pfor_control_flow_ops.pfor(loop_fn, 8)
162 return pfor_control_flow_ops.pfor(loop_fn, iters)
Darray_test.py30 from tensorflow.python.ops.parallel_for import control_flow_ops as pfor_control_flow_ops unknown
221 pfor_control_flow_ops.pfor(loop_fn, 2, fallback_to_while_loop=False)