Home
last modified time | relevance | path

Searched defs:bucket_id (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/hardware/
Ddevice_context.h118 …virtual std::shared_ptr<Bucket> CreateBucket(uint32_t bucket_id, uint32_t bucket_size) const { ret… in CreateBucket()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dunique_cpu_kernel.h85 auto bucket_id = BucketId(params->input_[i], bucket_num); in CalculateEachBucketSize() local
148 auto bucket_id = BucketId(data, segment->thread_num_); in SegmentToBuckets() local
Dsparse_optimizer_cpu_kernel.h156 auto bucket_id = index % bucket_num; in CalculateEachBucketSize() local
217 auto bucket_id = index % param.thread_num_; in CopySegmentIndicesToBucket() local
/third_party/boost/boost/pending/
Dbucket_sorter.hpp132 bucket_type bucket_id; member in boost::bucket_sorter::stack
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/gpu/
Dgpu_device_context.cc509 std::shared_ptr<Bucket> GPUDeviceContext::CreateBucket(uint32_t bucket_id, uint32_t bucket_size) co… in CreateBucket()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dsession_basic.h303 …virtual std::shared_ptr<device::Bucket> CreateBucket(uint32_t bucket_id, uint32_t bucket_size) { r… in CreateBucket()
Dgpu_session.cc737 std::shared_ptr<device::Bucket> GPUSession::CreateBucket(uint32_t bucket_id, uint32_t bucket_size) { in CreateBucket()
Dascend_session.cc1942 std::shared_ptr<device::Bucket> AscendSession::CreateBucket(uint32_t bucket_id, uint32_t bucket_siz… in CreateBucket()
Dsession_basic.cc2487 uint32_t bucket_id = 0; in InitAllBucket() local