Searched refs:reduced_bucket (Results 1 – 1 of 1) sorted by relevance
288 … const std::shared_ptr<SparseGradient<T>> &reduced_bucket) { in SortAndReduceBucketSparseGradient() argument293 MS_EXCEPTION_IF_NULL(reduced_bucket); in SortAndReduceBucketSparseGradient()294 MS_EXCEPTION_IF_NULL(reduced_bucket->value_); in SortAndReduceBucketSparseGradient()295 MS_EXCEPTION_IF_NULL(reduced_bucket->indices_); in SortAndReduceBucketSparseGradient()307 size_t max_length = reduced_bucket->indices_size_ * param.value_stride_; in SortAndReduceBucketSparseGradient()318 reduced_bucket->indices_[unique_indices_size] = index; in SortAndReduceBucketSparseGradient()320 …auto ret_code = memcpy_s(reduced_bucket->value_ + value_offset, (max_length - value_offset) * size… in SortAndReduceBucketSparseGradient()327 reduced_bucket->value_[value_offset + j] += global_value[global_value_offset + j]; in SortAndReduceBucketSparseGradient()332 reduced_bucket->indices_size_ = unique_indices_size; in SortAndReduceBucketSparseGradient()339 const std::shared_ptr<SparseGradient<T>> &reduced_bucket) { in ReduceBucketSparseGradient() argument[all …]