Home
last modified time | relevance | path

Searched defs:segment_bucket_sizes (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dsparse_optimizer_cpu_kernel.h108 std::vector<std::shared_ptr<std::vector<size_t>>> segment_bucket_sizes; in BucketReduceSparseGradient() local
172 auto &segment_bucket_sizes = *segment_bucket_sizes_ptr; in SplitAndCalculateSegmentBucketSize() local
232 const std::vector<std::shared_ptr<std::vector<size_t>>> &segment_bucket_sizes, in GatherSegmentIndicesToOutputBucket()
Dunique_cpu_kernel.h235 auto &segment_bucket_sizes = *segment_bucket_sizes_ptr; in SplitAndCalculateBucketSize() local
319 auto &segment_bucket_sizes = *segment_bucket_sizes_ptr; in GatherSegmentsToBuckets() local
509 std::vector<std::shared_ptr<std::vector<size_t>>> segment_bucket_sizes; in BucketUnique() local