Home
last modified time | relevance | path

Searched refs:bucket_index (Results 1 – 12 of 12) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/
Dbucket_batch_by_length_op.cc68 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()
Dbucket_batch_by_length_op.h70 Status PadAndBatchBucket(int32_t bucket_index, int32_t batch_size);
/third_party/mesa3d/src/gallium/auxiliary/pipebuffer/
Dpb_cache.c86 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()
Dpb_cache.h46 unsigned bucket_index; member
73 unsigned bucket_index);
76 struct pb_buffer *buf, unsigned bucket_index);
/third_party/node/deps/histogram/src/
Dhdr_histogram.c168 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/
Dpan_bo.c163 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/
Dimplementation.hpp3023 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/
Dlog_uniform_int_distribution_test.cc149 const auto bucket_index = [base_log, is_2, &param](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/
Dlog_uniform_int_distribution_test.cc149 const auto bucket_index = [base_log, is_2, &param](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/
Dlima_bo.c148 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/
Dunique_cpu_kernel.h149 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()
Dsparse_optimizer_cpu_kernel.h218 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()