Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/ops/parallel_for/
Dcontrol_flow_ops_test.py53 from tensorflow.python.ops.parallel_for import control_flow_ops as pfor_control_flow_ops unknown
95 pfor_control_flow_ops.pfor(lambda i: 1, 8, parallel_iterations=0)
97 pfor_control_flow_ops.for_loop(lambda i: 1, dtypes.int32, 8,
102 pfor_control_flow_ops.pfor(lambda i: 1, 8, parallel_iterations=1)
117 output = pfor_control_flow_ops.pfor(loop_fn, 8)
129 output = pfor_control_flow_ops.pfor(loop_fn, 8)
141 output = pfor_control_flow_ops.pfor(loop_fn, 8)
158 output = pfor_control_flow_ops.pfor(LoopFn(), 8)
172 output = pfor_control_flow_ops.pfor(loop_fn, 8)
186 pfor_control_flow_ops.pfor(loop_fn, 8, parallel_iterations=2)
[all …]
Dtest_util.py25 from tensorflow.python.ops.parallel_for import control_flow_ops as pfor_control_flow_ops unknown
55 t1 = pfor_control_flow_ops.pfor(loop_fn, iters=iters,
57 t2 = pfor_control_flow_ops.for_loop(loop_fn, loop_fn_dtypes, iters=iters,
Darray_test.py29 from tensorflow.python.ops.parallel_for import control_flow_ops as pfor_control_flow_ops unknown
141 pfor_control_flow_ops.pfor(loop_fn, 2)