Searched refs:pfor_config (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | control_flow_ops.py | 277 pfor_config=None, argument
288 if pfor_config is None:
289 pfor_config = PForConfig()
290 pfor_config._set_iters(iters) # pylint: disable=protected-access
291 loop_fn_outputs = loop_fn(loop_var, **{PFOR_CONFIG_ARG: pfor_config})
293 assert pfor_config is None
336 pfor_config=pfor_config,
343 … if pfor_config is not None and pfor_config._has_reductions(): # pylint: disable=protected-access
353 pfor_config=pfor_config)
367 def tiled_loop_fn(i, pfor_config=None): argument
[all …]
|
| D | xla_control_flow_ops_test.py | 114 def loop_fn(i, pfor_config): argument
116 return x_i - pfor_config.reduce_mean(x_i)
126 def _make_unstacked(cond, body, pfor_config): argument
129 return math_ops.reduce_any(pfor_config.reduce_concat(args[0]))
224 def loop_fn(i, pfor_config): argument
234 cond, body = _make_unstacked(_cond, _body, pfor_config)
|
| D | control_flow_ops_test.py | 241 def loop_fn(i, pfor_config): argument
244 reduced = pfor_config.reduce(reduce_fn, x_i, y_i)
255 def loop_fn(i, pfor_config): argument
257 vectorized_value = pfor_config.reduce_concat(x_i)
269 def loop_fn(i, pfor_config): argument
271 return x_i - pfor_config.reduce_mean(x_i)
281 def loop_fn(i, pfor_config): argument
283 return x_i - pfor_config.reduce_sum(x_i)
298 def __call__(self, i, pfor_config): argument
300 return x_i - pfor_config.reduce_mean(x_i)
[all …]
|
| D | pfor.py | 211 def __init__(self, exit_node, pfor_ops, fallback_to_while_loop, pfor_config): argument
222 self._pfor_config = pfor_config
405 pfor_config=self._pfor_config)
1279 pfor_config=None, argument
1321 self._pfor_config = pfor_config
1491 pfor_config=self._pfor_config)
1696 def pfor_config(self): member in PFor
4508 pfor_config=pfor.pfor_config)
4536 pfor_config=pfor_input.pfor.pfor_config)
4878 pfor_config=self._pfor.pfor_config)
[all …]
|