Searched defs:_batch_all_reduce (Results 1 – 1 of 1) sorted by relevance
694 def _batch_all_reduce(self, reduce_op, per_replica_values): member in AllReduceCrossDeviceOps878 def _batch_all_reduce(self, reduce_op, per_replica_values): member in MultiWorkerAllReduce1017 def _batch_all_reduce(self, reduce_op, per_replica_values): member in CollectiveAllReduce