Searched refs:pfor_control_flow_ops (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/python/ops/parallel_for/ |
D | control_flow_ops_test.py | 53 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 …]
|
D | test_util.py | 25 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,
|
D | array_test.py | 29 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)
|