Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/
H A Dcontrol_flow_ops.py140 fallback_to_while_loop=True, argument
196 fallback_to_while_loop=fallback_to_while_loop,
275 fallback_to_while_loop, argument
335 fallback_to_while_loop=fallback_to_while_loop,
352 fallback_to_while_loop=fallback_to_while_loop,
379 fallback_to_while_loop=fallback_to_while_loop,
440 def vectorized_map(fn, elems, fallback_to_while_loop=True, warn=True): argument
568 fallback_to_while_loop=fallback_to_while_loop,
H A Dtest_util.py52 fallback_to_while_loop=False, argument
58 fallback_to_while_loop=fallback_to_while_loop,
H A Dpfor.py211 def __init__(self, exit_node, pfor_ops, fallback_to_while_loop, pfor_config): argument
221 self._fallback_to_while_loop = fallback_to_while_loop
404 fallback_to_while_loop=self._fallback_to_while_loop,
1276 fallback_to_while_loop, argument
1319 self._fallback_to_while_loop = fallback_to_while_loop
1490 fallback_to_while_loop=self.fallback_to_while_loop,
1710 def fallback_to_while_loop(self): member in PFor
4505 fallback_to_while_loop=pfor.fallback_to_while_loop,
4533 fallback_to_while_loop=pfor_input.pfor.fallback_to_while_loop,
4874 fallback_to_while_loop=self._pfor.fallback_to_while_loop,
[all …]
H A Darray_test.py236 pfor_control_flow_ops.pfor(loop_fn, 2, fallback_to_while_loop=False)
496 self._test_loop_fn(loop_fn, 3, fallback_to_while_loop=True)
499 self._test_loop_fn(loop_fn, 3, fallback_to_while_loop=False)
H A Dcontrol_flow_ops_test.py97 self._test_loop_fn(loop_fn, 3, fallback_to_while_loop=False)
98 self._test_loop_fn(loop_fn, 3, fallback_to_while_loop=True)
1166 self._test_loop_fn(loop_fn, 3, fallback_to_while_loop=False)
1168 self._test_loop_fn(loop_fn, 3, fallback_to_while_loop=True)
2863 loop_fn, elems, fallback_to_while_loop=False)
2887 function, inputs, fallback_to_while_loop=False)
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dconstant_op_test.py127 vec_fn, constant_op.constant([1., 2.]), fallback_to_while_loop=False)
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.pbtxt1169 …argspec: "args=[\'fn\', \'elems\', \'fallback_to_while_loop\', \'warn\'], varargs=None, keywords=N…
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.pbtxt2525 …argspec: "args=[\'fn\', \'elems\', \'fallback_to_while_loop\', \'warn\'], varargs=None, keywords=N…