Searched refs:pfor_config (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | control_flow_ops.py | 270 pfor_config=None): argument
280 if pfor_config is None:
281 pfor_config = PForConfig()
282 pfor_config._set_iters(iters) # pylint: disable=protected-access
283 loop_fn_outputs = loop_fn(loop_var, **{PFOR_CONFIG_ARG: pfor_config})
285 assert pfor_config is None
321 pfor_config=pfor_config)
327 … if pfor_config is not None and pfor_config._has_reductions(): # pylint: disable=protected-access
337 pfor_config=pfor_config)
351 def tiled_loop_fn(i, pfor_config=None): argument
[all …]
|
| D | xla_control_flow_ops_test.py | 118 def loop_fn(i, pfor_config): argument
120 return x_i - pfor_config.reduce_mean(x_i)
130 def _make_unstacked(cond, body, pfor_config): argument
133 return math_ops.reduce_any(pfor_config.reduce_concat(args[0]))
225 def loop_fn(i, pfor_config): argument
235 cond, body = _make_unstacked(_cond, _body, pfor_config)
|
| D | control_flow_ops_test.py | 228 def loop_fn(i, pfor_config): argument
231 reduced = pfor_config.reduce(reduce_fn, x_i, y_i)
242 def loop_fn(i, pfor_config): argument
244 vectorized_value = pfor_config.reduce_concat(x_i)
256 def loop_fn(i, pfor_config): argument
258 return x_i - pfor_config.reduce_mean(x_i)
268 def loop_fn(i, pfor_config): argument
270 return x_i - pfor_config.reduce_sum(x_i)
285 def __call__(self, i, pfor_config): argument
287 return x_i - pfor_config.reduce_mean(x_i)
[all …]
|
| D | pfor.py | 204 def __init__(self, exit_node, pfor_ops, fallback_to_while_loop, pfor_config): argument
215 self._pfor_config = pfor_config
398 pfor_config=self._pfor_config)
1271 pfor_config=None): argument
1310 self._pfor_config = pfor_config
1480 pfor_config=self._pfor_config)
1678 def pfor_config(self): member in PFor
4191 pfor_config=pfor.pfor_config)
4219 pfor_config=pfor_input.pfor.pfor_config)
4429 pfor_config=self._pfor.pfor_config)
[all …]
|