Searched refs:reduce_to_device (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
208 with ops.device(reduce_to_device):
220 def _simple_gather(per_replica_value, reduce_to_device, axis): argument
226 with ops.device(reduce_to_device):
575 def __init__(self, reduce_to_device=None, accumulation_fn=None): argument
584 self.reduce_to_device = reduce_to_device
595 reduce_to_device = self.reduce_to_device or devices[0]
598 "Reduce to %s then broadcast to %r." % (reduce_to_device, devices), 10)
599 reduced = _simple_reduce(per_replica_value, reduce_to_device,
610 reduce_to_device = self.reduce_to_device or devices[0]
[all …]
|
| D | parameter_server_strategy.py | 178 cross_device_ops_lib.ReductionToOneDevice(reduce_to_device=_LOCAL_CPU))
|
| /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 | 288 reduce_to_device=_cpu_device)),
|