Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dunique_cpu_kernel.h229 … std::vector<std::shared_ptr<std::vector<size_t>>> *segment_bucket_sizes_ptr) { in SplitAndCalculateBucketSize()
306 … std::vector<std::shared_ptr<std::vector<size_t>>> *segment_bucket_sizes_ptr, in GatherSegmentsToBuckets()
Dsparse_optimizer_cpu_kernel.h167 std::vector<std::shared_ptr<std::vector<size_t>>> *segment_bucket_sizes_ptr) { in SplitAndCalculateSegmentBucketSize()