| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops.py | 204 def _simple_reduce(per_replica_value, reduce_to_device, accumulation_fn, argument
207 all_values = per_replica_value.values
231 def reduce(self, reduce_op, per_replica_value, destinations): argument
250 if not isinstance(per_replica_value, value_lib.PerReplica):
251 per_replica_value = _make_tensor_into_per_replica(per_replica_value)
254 return self.reduce_implementation(reduce_op, per_replica_value,
303 def reduce_implementation(self, reduce_op, per_replica_value, destinations): argument
382 def reduce_implementation(self, reduce_op, per_replica_value, destinations): argument
386 devices = get_devices_from(per_replica_value)
391 reduced = _simple_reduce(per_replica_value, reduce_to_device,
[all …]
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.-cross-device-ops.pbtxt | 27 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\'], varargs=None, k…
31 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\'], varargs=None, k…
|
| D | tensorflow.distribute.-nccl-all-reduce.pbtxt | 29 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\'], varargs=None, k…
33 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\'], varargs=None, k…
|
| D | tensorflow.distribute.-reduction-to-one-device.pbtxt | 28 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\'], varargs=None, k…
32 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\'], varargs=None, k…
|
| D | tensorflow.distribute.-hierarchical-copy-all-reduce.pbtxt | 29 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\'], varargs=None, k…
33 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\'], varargs=None, k…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-cross-device-ops.pbtxt | 27 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\'], varargs=None, k…
31 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\'], varargs=None, k…
|
| D | tensorflow.distribute.-nccl-all-reduce.pbtxt | 29 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\'], varargs=None, k…
33 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\'], varargs=None, k…
|
| D | tensorflow.distribute.-reduction-to-one-device.pbtxt | 28 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\'], varargs=None, k…
32 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\'], varargs=None, k…
|
| D | tensorflow.distribute.-hierarchical-copy-all-reduce.pbtxt | 29 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\'], varargs=None, k…
33 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\'], varargs=None, k…
|