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