Searched defs:pfor_config (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | xla_control_flow_ops_test.py | 118 def loop_fn(i, pfor_config): argument
130 def _make_unstacked(cond, body, pfor_config): argument
228 def loop_fn(i, pfor_config): argument
|
| D | control_flow_ops.py | 270 pfor_config=None): argument
351 def tiled_loop_fn(i, pfor_config=None): argument
|
| D | control_flow_ops_test.py | 245 def loop_fn(i, pfor_config): argument
259 def loop_fn(i, pfor_config): argument
273 def loop_fn(i, pfor_config): argument
285 def loop_fn(i, pfor_config): argument
302 def __call__(self, i, pfor_config): argument
314 def fn(i, pfor_config, dummy=None): argument
328 def loop_fn(i, pfor_config): argument
342 def loop_fn(i, pfor_config): argument
2245 def loop_fn(i, pfor_config): argument
|
| D | pfor.py | 215 def __init__(self, exit_node, pfor_ops, fallback_to_while_loop, pfor_config): argument
1282 pfor_config=None): argument
1697 def pfor_config(self): member in PFor
|