Searched refs:accumulation_fn (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops.py | 204 def _simple_reduce(per_replica_value, reduce_to_device, accumulation_fn, argument
215 all_values, accumulation_fn)
369 def __init__(self, reduce_to_device=None, accumulation_fn=None): argument
379 self.accumulation_fn = accumulation_fn or math_ops.add_n
392 self.accumulation_fn, reduce_op)
|
| D | cross_device_utils.py | 645 def aggregate_tensors_or_indexed_slices(values, accumulation_fn=math_ops.add_n): argument
650 return accumulation_fn(values)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.-reduction-to-one-device.pbtxt | 8 …argspec: "args=[\'self\', \'reduce_to_device\', \'accumulation_fn\'], varargs=None, keywords=None,…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-reduction-to-one-device.pbtxt | 8 …argspec: "args=[\'self\', \'reduce_to_device\', \'accumulation_fn\'], varargs=None, keywords=None,…
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | cross_device_ops_test.py | 217 accumulation_fn=math_ops.accumulate_n)),
|