Searched refs:thread_buckets (Results 1 – 2 of 2) sorted by relevance
260 std::vector<std::shared_ptr<BucketSparseGradient<T>>> thread_buckets; in GatherSegmentIndicesToOutputBucket() local262 (void)thread_buckets.emplace_back(std::make_shared<BucketSparseGradient<T>>()); in GatherSegmentIndicesToOutputBucket()263 thread_buckets[j]->indices_ = buckets[j]->indices_ + tmp_bucket_data_size[j]; in GatherSegmentIndicesToOutputBucket()264 thread_buckets[j]->global_indices_ = buckets[j]->global_indices_ + tmp_bucket_data_size[j]; in GatherSegmentIndicesToOutputBucket()265 … thread_buckets[j]->value_ = buckets[j]->value_ + tmp_bucket_data_size[j] * param.value_stride_; in GatherSegmentIndicesToOutputBucket()266 thread_buckets[j]->indices_size_ = segment_bucket_sizes[i]->at(j); in GatherSegmentIndicesToOutputBucket()269 (void)each_thread_buckets.emplace_back(thread_buckets); in GatherSegmentIndicesToOutputBucket()
207 std::vector<std::vector<std::shared_ptr<UniqueParam<DataType, IndexType>>>> thread_buckets; in GatherSegmentsToBuckets() local218 (void)thread_buckets.emplace_back(local_buckets); in GatherSegmentsToBuckets()225 auto task = [&segments, &thread_buckets, current_offset, i]() { in GatherSegmentsToBuckets()226 SegmentToBuckets<DataType, IndexType>(segments[i], current_offset, thread_buckets[i]); in GatherSegmentsToBuckets()