Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dunique_cpu_kernel.h228 … std::vector<std::shared_ptr<UniqueParam<DataType, IndexType>>> *segments_ptr, in SplitAndCalculateBucketSize()
305 … std::vector<std::shared_ptr<UniqueParam<DataType, IndexType>>> *segments_ptr, in GatherSegmentsToBuckets()
Dsparse_optimizer_cpu_kernel.h166 …ltiThreadReduceSparseGradientParam<T> &param, std::vector<std::shared_ptr<SparseGradient<T>>> *seg… in SplitAndCalculateSegmentBucketSize()