Searched refs:data_sort (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Bucketization.cu | 26 …(const input_t *data_ss, int64_t start, int64_t end, const input_t val, const int64_t *data_sort) { in lower_bound() argument
32 const input_t mid_val = data_sort ? data_ss[orig_start + data_sort[mid]] : data_ss[mid]; in lower_bound()
44 …(const input_t *data_ss, int64_t start, int64_t end, const input_t val, const int64_t *data_sort) { in upper_bound() argument
50 const input_t mid_val = data_sort ? data_ss[orig_start + data_sort[mid]] : data_ss[mid]; in upper_bound()
66 const int64_t *data_sort, in searchsorted_cuda_kernel() argument
79 lower_bound<input_t>(data_bd, start_bd, end_bd, data_in[tid], data_sort) - start_bd : in searchsorted_cuda_kernel()
80 upper_bound<input_t>(data_bd, start_bd, end_bd, data_in[tid], data_sort) - start_bd; in searchsorted_cuda_kernel()
97 const int64_t *data_sort = sorter.defined() ? sorter.const_data_ptr<int64_t>() : nullptr; in searchsorted_cuda_contiguous() local
107 … data_out, data_in, data_bd, data_sort, idim_in, idim_bd, numel_in, right, boundaries.dim() == 1); in searchsorted_cuda_contiguous()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/operations/ |
| H A D | Bucketization.mm | 28 …ant input_t *data_ss, int64_t start, int64_t end, const input_t val, constant int64_t *data_sort) {
34 const input_t mid_val = data_ss[orig_start + data_sort[mid]];
61 …ant input_t *data_ss, int64_t start, int64_t end, const input_t val, constant int64_t *data_sort) {
67 const input_t mid_val = data_ss[orig_start + data_sort[mid]];
103 constant int64_t * data_sort [[buffer(8)]],
114 lower_bound<input_t>(data_bd, start_bd, end_bd, data_in[tid], data_sort) - start_bd :
115 upper_bound<input_t>(data_bd, start_bd, end_bd, data_in[tid], data_sort) - start_bd;
162 constant int64_t * data_sort [[buffer(8)]], \
|