/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/ |
D | bucket_batch_by_length_op.cc | 68 int bucket_index = bucket_boundaries_.size() - 1; in operator ()() local 69 while (element_length < bucket_boundaries_[bucket_index]) { in operator ()() 70 bucket_index--; in operator ()() 73 buckets_[bucket_index]->push_back(current_row); in operator ()() 75 if (buckets_[bucket_index]->size() == bucket_batch_sizes_[bucket_index]) { in operator ()() 76 RETURN_IF_NOT_OK(PadAndBatchBucket(bucket_index, bucket_batch_sizes_[bucket_index])); in operator ()() 130 Status BucketBatchByLengthOp::PadAndBatchBucket(int32_t bucket_index, int32_t batch_size) { in PadAndBatchBucket() argument 131 std::unique_ptr<TensorQTable> *bucket = &buckets_[bucket_index]; in PadAndBatchBucket() 140 if (bucket_index + 1 >= bucket_boundaries_.size()) { in PadAndBatchBucket() 146 pad_shape[i] = bucket_boundaries_[bucket_index + 1] - 1; in PadAndBatchBucket()
|
D | bucket_batch_by_length_op.h | 70 Status PadAndBatchBucket(int32_t bucket_index, int32_t batch_size);
|
/third_party/mesa3d/src/gallium/auxiliary/pipebuffer/ |
D | pb_cache.c | 86 struct list_head *cache = &mgr->buckets[entry->bucket_index]; in pb_cache_add_buffer() 149 unsigned bucket_index) in pb_cache_reclaim_buffer() argument 157 assert(bucket_index < mgr->num_heaps); in pb_cache_reclaim_buffer() 158 struct list_head *cache = &mgr->buckets[bucket_index]; in pb_cache_reclaim_buffer() 251 struct pb_buffer *buf, unsigned bucket_index) in pb_cache_init_entry() argument 253 assert(bucket_index < mgr->num_heaps); in pb_cache_init_entry() 258 entry->bucket_index = bucket_index; in pb_cache_init_entry()
|
D | pb_cache.h | 46 unsigned bucket_index; member 73 unsigned bucket_index); 76 struct pb_buffer *buf, unsigned bucket_index);
|
/third_party/node/deps/histogram/src/ |
D | hdr_histogram.c | 168 static int32_t get_sub_bucket_index(int64_t value, int32_t bucket_index, int32_t unit_magnitude) in get_sub_bucket_index() argument 170 return (int32_t)(value >> (bucket_index + unit_magnitude)); in get_sub_bucket_index() 173 static int32_t counts_index(const struct hdr_histogram* h, int32_t bucket_index, int32_t sub_bucket… in counts_index() argument 177 int32_t bucket_base_index = (bucket_index + 1) << h->sub_bucket_half_count_magnitude; in counts_index() 184 static int64_t value_from_index(int32_t bucket_index, int32_t sub_bucket_index, int32_t unit_magnit… in value_from_index() argument 186 return ((int64_t) sub_bucket_index) << (bucket_index + unit_magnitude); in value_from_index() 191 int32_t bucket_index = get_bucket_index(h, value); in counts_index_for() local 192 int32_t sub_bucket_index = get_sub_bucket_index(value, bucket_index, h->unit_magnitude); in counts_index_for() 194 return counts_index(h, bucket_index, sub_bucket_index); in counts_index_for() 199 int32_t bucket_index = (index >> h->sub_bucket_half_count_magnitude) - 1; in hdr_value_at_index() local [all …]
|
/third_party/mesa3d/src/panfrost/lib/ |
D | pan_bo.c | 163 unsigned bucket_index = util_logbase2(size); in pan_bucket_index() local 168 bucket_index = MIN2(bucket_index, MAX_BO_CACHE_BUCKET); in pan_bucket_index() 173 assert(bucket_index >= MIN_BO_CACHE_BUCKET); in pan_bucket_index() 174 assert(bucket_index <= MAX_BO_CACHE_BUCKET); in pan_bucket_index() 177 return (bucket_index - MIN_BO_CACHE_BUCKET); in pan_bucket_index()
|
/third_party/boost/boost/unordered/detail/ |
D | implementation.hpp | 3023 bucket_pointer get_bucket_pointer(std::size_t bucket_index) const in get_bucket_pointer() 3026 return buckets_ + static_cast<std::ptrdiff_t>(bucket_index); in get_bucket_pointer() 3034 link_pointer get_previous_start(std::size_t bucket_index) const in get_previous_start() 3036 return get_bucket_pointer(bucket_index)->next_; in get_previous_start() 3044 node_pointer begin(std::size_t bucket_index) const in begin() 3048 link_pointer prev = get_previous_start(bucket_index); in begin() 3388 std::size_t bucket_index, link_pointer prev, node_pointer next) in fix_bucket() 3390 std::size_t bucket_index2 = bucket_index; in fix_bucket() 3396 if (bucket_index == bucket_index2) { in fix_bucket() 3405 bucket_pointer this_bucket = get_bucket_pointer(bucket_index); in fix_bucket() [all …]
|
/third_party/skia/third_party/externals/abseil-cpp/absl/random/ |
D | log_uniform_int_distribution_test.cc | 149 const auto bucket_index = [base_log, is_2, ¶m](int32_t x) { in ChiSquaredTestImpl() local 155 const int max_bucket = bucket_index(param.max()); // inclusive in ChiSquaredTestImpl() 167 int bucket = bucket_index(sample); in ChiSquaredTestImpl()
|
/third_party/abseil-cpp/absl/random/ |
D | log_uniform_int_distribution_test.cc | 149 const auto bucket_index = [base_log, is_2, ¶m](int32_t x) { in ChiSquaredTestImpl() local 155 const int max_bucket = bucket_index(param.max()); // inclusive in ChiSquaredTestImpl() 167 int bucket = bucket_index(sample); in ChiSquaredTestImpl()
|
/third_party/mesa3d/src/gallium/drivers/lima/ |
D | lima_bo.c | 148 unsigned bucket_index = util_logbase2(size); in lima_bucket_index() local 152 bucket_index = CLAMP(bucket_index, MIN_BO_CACHE_BUCKET, in lima_bucket_index() 156 return (bucket_index - MIN_BO_CACHE_BUCKET); in lima_bucket_index()
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
D | unique_cpu_kernel.h | 149 auto bucket_index = bucket_data_num[bucket_id]; in SegmentToBuckets() local 156 if (bucket_index >= bucket->input_size_) { in SegmentToBuckets() 160 bucket->input_[bucket_index] = data; in SegmentToBuckets() 161 bucket->workspace_idx_[bucket_index] = segment_offset + i; in SegmentToBuckets()
|
D | sparse_optimizer_cpu_kernel.h | 218 auto bucket_index = bucket_data_num[bucket_id]; in CopySegmentIndicesToBucket() local 219 buckets[bucket_id]->indices_[bucket_index] = index; in CopySegmentIndicesToBucket() 220 buckets[bucket_id]->global_indices_[bucket_index] = bucket_offset + i; in CopySegmentIndicesToBucket()
|