Searched defs:reduce_implementation (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | cross_device_ops.py | 466 def reduce_implementation(self, reduce_op, per_replica_value, destinations, member in CrossDeviceOps
608 def reduce_implementation(self, reduce_op, per_replica_value, destinations, member in ReductionToOneDevice
856 def reduce_implementation(self, reduce_op, per_replica_value, destinations, member in AllReduceCrossDeviceOps
1220 def reduce_implementation(self, reduce_op, per_replica_value, destinations, member in CollectiveAllReduce
|