Searched refs:_do_batch_all_reduce_sparse (Results 1 – 1 of 1) sorted by relevance
703 sparse_results = self._do_batch_all_reduce_sparse(reduce_op,743 def _do_batch_all_reduce_sparse(self, reduce_op, sparse_values): member in AllReduceCrossDeviceOps