27  * - torch.bucketize(values, boundaries, right=False, out_int32=False)
29  *   boundaries - 1D tensor containing a sorted sequences
88 void searchsorted_cpu_contiguous(Tensor& result, const Tensor& input, const Tensor& boundaries, con…  in searchsorted_cpu_contiguous()  argument
91   // inner most dim size of input and boundaries  in searchsorted_cpu_contiguous()
93   int64_t idim_bd = boundaries.sizes().back();  in searchsorted_cpu_contiguous()
96   const input_t *data_bd = boundaries.const_data_ptr<input_t>();  in searchsorted_cpu_contiguous()
100   bool is_1d_boundaries = boundaries.dim() == 1;  in searchsorted_cpu_contiguous()
103       // If boundaries tensor is 1d, we always search the entire boundary tensor  in searchsorted_cpu_contiguous()
117 void dispatch(Tensor& result, const Tensor& input, const Tensor& boundaries, bool out_int32, bool r…  in dispatch()  argument
126               result, input, boundaries, right, sorter);  in dispatch()
137               result, input, boundaries, right, sorter);  in dispatch()
228 Tensor& bucketize_out_cpu(const Tensor& self, const Tensor& boundaries, bool out_int32, bool right,…  in bucketize_out_cpu()  argument
229 …TORCH_CHECK(boundaries.dim() == 1, "boundaries tensor must be 1 dimension, but got dim(", boundari…  in bucketize_out_cpu()
230 …at::native::searchsorted_out_cpu(boundaries, self, out_int32, right, std::nullopt, std::nullopt, r…  in bucketize_out_cpu()
234 Tensor bucketize_cpu(const Tensor& self, const Tensor& boundaries, bool out_int32, bool right) {  in bucketize_cpu()  argument
238   at::native::bucketize_out_cpu(self, boundaries, out_int32, right, result);  in bucketize_cpu()
242 Tensor bucketize_cpu(const Scalar& self, const Tensor& boundaries, bool out_int32, bool right) {  in bucketize_cpu()  argument
243 …return bucketize_cpu(searchsorted_scalar_tensor(self, boundaries.device()), boundaries, out_int32,…  in bucketize_cpu()