Searched refs:_do_batch_all_reduce_sparse (Results 1 – 1 of 1) sorted by relevance
868 sparse_results = self._do_batch_all_reduce_sparse(reduce_op,906 def _do_batch_all_reduce_sparse(self, reduce_op, sparse_values): member in AllReduceCrossDeviceOps1191 sparse_results = self._do_batch_all_reduce_sparse(reduce_op,1266 def _do_batch_all_reduce_sparse(self, reduce_op, per_replica_values, options): member in CollectiveAllReduce