Searched defs:_batch_all_reduce (Results 1 – 1 of 1) sorted by relevance
859 def _batch_all_reduce(self, reduce_op, per_replica_values): member in AllReduceCrossDeviceOps1181 def _batch_all_reduce(self, reduce_op, per_replica_values, options): member in CollectiveAllReduce