Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dunique_cpu_kernel.h93 std::vector<std::shared_ptr<UniqueParam<DataType, IndexType>>> *segments_ptr, in SplitAndCalculateBucketSize() argument
97 MS_EXCEPTION_IF_NULL(segments_ptr); in SplitAndCalculateBucketSize()
99 auto &segments = *segments_ptr; in SplitAndCalculateBucketSize()
169 … std::vector<std::shared_ptr<UniqueParam<DataType, IndexType>>> *segments_ptr, in GatherSegmentsToBuckets() argument
179 MS_EXCEPTION_IF_NULL(segments_ptr); in GatherSegmentsToBuckets()
182 auto &segments = *segments_ptr; in GatherSegmentsToBuckets()
Dsparse_optimizer_cpu_kernel.h165 …ReduceSparseGradientParam<T> &param, std::vector<std::shared_ptr<SparseGradient<T>>> *segments_ptr, in SplitAndCalculateSegmentBucketSize() argument
169 MS_EXCEPTION_IF_NULL(segments_ptr); in SplitAndCalculateSegmentBucketSize()
170 auto &segments = *segments_ptr; in SplitAndCalculateSegmentBucketSize()