| H A D | cross_device_ops.py | 317 return self.reduce_implementation(reduce_op, per_replica_value,
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
640 self.reduce_implementation(
856 def reduce_implementation(self, reduce_op, per_replica_value, destinations, member in AllReduceCrossDeviceOps
875 self.reduce_implementation(reduce_op, value, dest, options)
1220 def reduce_implementation(self, reduce_op, per_replica_value, destinations, member in CollectiveAllReduce
1269 self.reduce_implementation(reduce_op, value, dest, options)
|