Home
last modified time | relevance | path

Searched refs:segment_bucket_sizes_ptr (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dunique_cpu_kernel.h94 std::vector<std::shared_ptr<std::vector<IndexType>>> *segment_bucket_sizes_ptr) { in SplitAndCalculateBucketSize() argument
98 MS_EXCEPTION_IF_NULL(segment_bucket_sizes_ptr); in SplitAndCalculateBucketSize()
100 auto &segment_bucket_sizes = *segment_bucket_sizes_ptr; in SplitAndCalculateBucketSize()
170 … std::vector<std::shared_ptr<std::vector<IndexType>>> *segment_bucket_sizes_ptr, in GatherSegmentsToBuckets() argument
180 MS_EXCEPTION_IF_NULL(segment_bucket_sizes_ptr); in GatherSegmentsToBuckets()
183 auto &segment_bucket_sizes = *segment_bucket_sizes_ptr; in GatherSegmentsToBuckets()
Dsparse_optimizer_cpu_kernel.h166 std::vector<std::shared_ptr<std::vector<size_t>>> *segment_bucket_sizes_ptr) { in SplitAndCalculateSegmentBucketSize() argument
168 MS_EXCEPTION_IF_NULL(segment_bucket_sizes_ptr); in SplitAndCalculateSegmentBucketSize()
171 auto &segment_bucket_sizes = *segment_bucket_sizes_ptr; in SplitAndCalculateSegmentBucketSize()