Searched defs:batch_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 | 498 def batch_reduce_implementation(self, reduce_op, value_destination_pairs, member in CrossDeviceOps
637 def batch_reduce_implementation(self, reduce_op, value_destination_pairs, member in ReductionToOneDevice
868 def batch_reduce_implementation(self, reduce_op, value_destination_pairs, member in AllReduceCrossDeviceOps
1254 def batch_reduce_implementation(self, reduce_op, value_destination_pairs, member in CollectiveAllReduce
|