| /third_party/boost/boost/pending/ |
| D | bucket_sorter.hpp | 98 : bucket_id(_bucket_id), head(h), next(n), prev(p), value(v), id(_id) in stack()
105 : bucket_id(_bucket_id), head(h), next(n), prev(p), value(v) in stack()
112 const size_type current = head[bucket_id]; in push()
117 head[bucket_id] = new_head; in push()
121 size_type current = head[bucket_id]; in pop()
123 head[bucket_id] = next_node; in pop()
127 value_type& top() { return value[head[bucket_id]]; } in top()
128 const value_type& top() const { return value[head[bucket_id]]; } in top()
129 bool empty() const { return head[bucket_id] == invalid_value(); } in empty()
132 bucket_type bucket_id; member in boost::bucket_sorter::stack
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | unique_cpu_kernel.h | 85 auto bucket_id = BucketId(params->input_[i], bucket_num); in CalculateEachBucketSize() local
86 each_bucket_size->at(bucket_id)++; in CalculateEachBucketSize()
148 auto bucket_id = BucketId(data, segment->thread_num_); in SegmentToBuckets() local
149 auto bucket_index = bucket_data_num[bucket_id]; in SegmentToBuckets()
150 if (bucket_id >= bucket_size) { in SegmentToBuckets()
154 auto &bucket = buckets[bucket_id]; in SegmentToBuckets()
162 bucket_data_num[bucket_id]++; in SegmentToBuckets()
|
| D | sparse_optimizer_cpu_kernel.h | 156 auto bucket_id = index % bucket_num; in CalculateEachBucketSize() local
157 each_bucket_size->at(bucket_id)++; in CalculateEachBucketSize()
217 auto bucket_id = index % param.thread_num_; in CopySegmentIndicesToBucket() local
218 auto bucket_index = bucket_data_num[bucket_id]; in CopySegmentIndicesToBucket()
219 buckets[bucket_id]->indices_[bucket_index] = index; in CopySegmentIndicesToBucket()
220 buckets[bucket_id]->global_indices_[bucket_index] = bucket_offset + i; in CopySegmentIndicesToBucket()
221 bucket_data_num[bucket_id]++; in CopySegmentIndicesToBucket()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hardware/ |
| D | device_context.h | 118 …virtual std::shared_ptr<Bucket> CreateBucket(uint32_t bucket_id, uint32_t bucket_size) const { ret… in CreateBucket() argument
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hardware/gpu/ |
| D | gpu_device_context.h | 72 std::shared_ptr<Bucket> CreateBucket(uint32_t bucket_id, uint32_t bucket_size) const override;
|
| D | gpu_device_context.cc | 509 std::shared_ptr<Bucket> GPUDeviceContext::CreateBucket(uint32_t bucket_id, uint32_t bucket_size) co… in CreateBucket() argument
510 auto bucket = std::make_shared<GPUBucket>(bucket_id, bucket_size); in CreateBucket()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | gpu_session.h | 56 std::shared_ptr<device::Bucket> CreateBucket(uint32_t bucket_id, uint32_t bucket_size) override;
|
| D | ascend_session.h | 158 std::shared_ptr<device::Bucket> CreateBucket(uint32_t bucket_id, uint32_t bucket_size) override;
|
| D | gpu_session.cc | 737 std::shared_ptr<device::Bucket> GPUSession::CreateBucket(uint32_t bucket_id, uint32_t bucket_size) { in CreateBucket() argument
738 auto bucket = std::make_shared<device::gpu::GPUBucket>(bucket_id, bucket_size); in CreateBucket()
|
| D | session_basic.h | 303 …virtual std::shared_ptr<device::Bucket> CreateBucket(uint32_t bucket_id, uint32_t bucket_size) { r… in CreateBucket() argument
|
| D | session_basic.cc | 2487 uint32_t bucket_id = 0; in InitAllBucket() local
2489 MS_LOG(INFO) << "Create new bucket:" << bucket_id << " size:" << bucket_size; in InitAllBucket()
2492 bucket = device_context->CreateBucket(bucket_id++, bucket_size); in InitAllBucket()
2494 bucket = CreateBucket(bucket_id++, bucket_size); in InitAllBucket()
|
| D | ascend_session.cc | 1942 std::shared_ptr<device::Bucket> AscendSession::CreateBucket(uint32_t bucket_id, uint32_t bucket_siz… in CreateBucket() argument
1943 auto bucket = std::make_shared<device::ascend::AscendBucket>(bucket_id, bucket_size); in CreateBucket()
|