Searched refs:BucketSparseGradient (Results 1 – 1 of 1) sorted by relevance
72 struct BucketSparseGradient { struct110 std::vector<std::shared_ptr<BucketSparseGradient<T>>> buckets; in BucketReduceSparseGradient()206 … const std::vector<std::shared_ptr<BucketSparseGradient<T>>> &buckets) { in CopySegmentIndicesToBucket()232 std::vector<std::shared_ptr<BucketSparseGradient<T>>> *buckets_ptr) { in GatherSegmentIndicesToOutputBucket()250 (void)buckets.emplace_back(std::make_shared<BucketSparseGradient<T>>()); in GatherSegmentIndicesToOutputBucket()258 std::vector<std::vector<std::shared_ptr<BucketSparseGradient<T>>>> each_thread_buckets; in GatherSegmentIndicesToOutputBucket()260 std::vector<std::shared_ptr<BucketSparseGradient<T>>> thread_buckets; in GatherSegmentIndicesToOutputBucket()262 (void)thread_buckets.emplace_back(std::make_shared<BucketSparseGradient<T>>()); in GatherSegmentIndicesToOutputBucket()287 … const std::shared_ptr<BucketSparseGradient<T>> &bucket, in SortAndReduceBucketSparseGradient()338 const std::shared_ptr<BucketSparseGradient<T>> &bucket, in ReduceBucketSparseGradient()[all …]