Searched refs:each_thread_buckets (Results 1 – 1 of 1) sorted by relevance
258 std::vector<std::vector<std::shared_ptr<BucketSparseGradient<T>>>> each_thread_buckets; in GatherSegmentIndicesToOutputBucket() local269 (void)each_thread_buckets.emplace_back(thread_buckets); in GatherSegmentIndicesToOutputBucket()275 auto task = [¶m, &segments, &each_thread_buckets, i, current_indices_offset]() { in GatherSegmentIndicesToOutputBucket()276 … CopySegmentIndicesToBucket<T>(param, segments[i], current_indices_offset, each_thread_buckets[i]); in GatherSegmentIndicesToOutputBucket()