Home
last modified time | relevance | path

Searched refs:fallback_to_while_loop (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/ops/parallel_for/
Dcontrol_flow_ops.py138 def pfor(loop_fn, iters, fallback_to_while_loop=True, parallel_iterations=None): argument
190 fallback_to_while_loop=fallback_to_while_loop,
268 fallback_to_while_loop, argument
320 fallback_to_while_loop=fallback_to_while_loop,
336 fallback_to_while_loop=fallback_to_while_loop,
363 fallback_to_while_loop=fallback_to_while_loop,
424 def vectorized_map(fn, elems, fallback_to_while_loop=True): argument
547 fallback_to_while_loop=fallback_to_while_loop)
Dtest_util.py56 fallback_to_while_loop=False, argument
62 fallback_to_while_loop=fallback_to_while_loop,
Darray_test.py232 pfor_control_flow_ops.pfor(loop_fn, 2, fallback_to_while_loop=False)
482 self._test_loop_fn(loop_fn, 3, fallback_to_while_loop=True)
485 self._test_loop_fn(loop_fn, 3, fallback_to_while_loop=False)
Dpfor.py215 def __init__(self, exit_node, pfor_ops, fallback_to_while_loop, pfor_config): argument
225 self._fallback_to_while_loop = fallback_to_while_loop
408 fallback_to_while_loop=self._fallback_to_while_loop,
1279 fallback_to_while_loop, argument
1320 self._fallback_to_while_loop = fallback_to_while_loop
1490 fallback_to_while_loop=self.fallback_to_while_loop,
1711 def fallback_to_while_loop(self): member in PFor
4392 fallback_to_while_loop=pfor.fallback_to_while_loop,
4420 fallback_to_while_loop=pfor_input.pfor.fallback_to_while_loop,
4715 fallback_to_while_loop=self._pfor.fallback_to_while_loop,
[all …]
Dcontrol_flow_ops_test.py101 self._test_loop_fn(loop_fn, 3, fallback_to_while_loop=False)
102 self._test_loop_fn(loop_fn, 3, fallback_to_while_loop=True)
1173 self._test_loop_fn(loop_fn, 3, fallback_to_while_loop=False)
1175 self._test_loop_fn(loop_fn, 3, fallback_to_while_loop=True)
/external/tensorflow/tensorflow/python/framework/
Dconstant_op_test.py131 vec_fn, constant_op.constant([1., 2.]), fallback_to_while_loop=False)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.pbtxt1169 …argspec: "args=[\'fn\', \'elems\', \'fallback_to_while_loop\'], varargs=None, keywords=None, defau…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.pbtxt2521 …argspec: "args=[\'fn\', \'elems\', \'fallback_to_while_loop\'], varargs=None, keywords=None, defau…