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
225 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 | 228 def loop_fn(i, pfor_config): argument
242 def loop_fn(i, pfor_config): argument
256 def loop_fn(i, pfor_config): argument
268 def loop_fn(i, pfor_config): argument
285 def __call__(self, i, pfor_config): argument
297 def fn(i, pfor_config, dummy=None): argument
311 def loop_fn(i, pfor_config): argument
325 def loop_fn(i, pfor_config): argument
1952 def loop_fn(i, pfor_config): argument
|
| D | pfor.py | 204 def __init__(self, exit_node, pfor_ops, fallback_to_while_loop, pfor_config): argument
1271 pfor_config=None): argument
1678 def pfor_config(self): member in PFor
|