Searched refs:sparse_grad (Results 1 – 1 of 1) sorted by relevance
143 …static void CalculateEachBucketSize(const std::shared_ptr<SparseGradient<T>> &sparse_grad, size_t … in CalculateEachBucketSize() argument146 MS_EXCEPTION_IF_NULL(sparse_grad); in CalculateEachBucketSize()147 MS_EXCEPTION_IF_NULL(sparse_grad->indices_); in CalculateEachBucketSize()153 for (size_t i = 0; i < sparse_grad->indices_size_; ++i) { in CalculateEachBucketSize()154 T index = sparse_grad->indices_[i]; in CalculateEachBucketSize()