Searched defs:pfor_config (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/ |
| H A D | xla_control_flow_ops_test.py | 114 def loop_fn(i, pfor_config): argument
126 def _make_unstacked(cond, body, pfor_config): argument
224 def loop_fn(i, pfor_config): argument
|
| H A D | control_flow_ops_test.py | 241 def loop_fn(i, pfor_config): argument
255 def loop_fn(i, pfor_config): argument
269 def loop_fn(i, pfor_config): argument
281 def loop_fn(i, pfor_config): argument
298 def __call__(self, i, pfor_config): argument
310 def fn(i, pfor_config, dummy=None): argument
324 def loop_fn(i, pfor_config): argument
338 def loop_fn(i, pfor_config): argument
2326 def loop_fn(i, pfor_config): argument
|
| H A D | control_flow_ops.py | 277 pfor_config=None, argument
367 def tiled_loop_fn(i, pfor_config=None): argument
|
| H A D | pfor.py | 211 def __init__(self, exit_node, pfor_ops, fallback_to_while_loop, pfor_config): argument
1279 pfor_config=None, argument
1696 def pfor_config(self): member in PFor
|