Searched refs:accumulation_fn (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops.py | 200 def _simple_reduce(per_replica_value, reduce_to_device, accumulation_fn, argument
211 all_values, accumulation_fn)
575 def __init__(self, reduce_to_device=None, accumulation_fn=None): argument
585 self.accumulation_fn = accumulation_fn or math_ops.add_n
600 self.accumulation_fn, reduce_op)
|
| D | cross_device_utils.py | 574 def aggregate_tensors_or_indexed_slices(values, accumulation_fn=math_ops.add_n): argument
579 return accumulation_fn(values)
|
| /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/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/python/distribute/v1/ |
| D | cross_device_ops_test.py | 292 accumulation_fn=math_ops.add_n)),
|