Searched refs:searchsorted_scalar_tensor (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Bucketization.cpp | 199 const Tensor& scalar_tensor = searchsorted_scalar_tensor(self, sorted_sequence.device()); in searchsorted_out_cpu()
224 const Tensor& scalar_tensor = searchsorted_scalar_tensor(self, sorted_sequence.device()); in searchsorted_cpu()
243 …return bucketize_cpu(searchsorted_scalar_tensor(self, boundaries.device()), boundaries, out_int32,… in bucketize_cpu()
|
| H A D | BucketizationUtils.h | 96 inline Tensor searchsorted_scalar_tensor(const Scalar& scalar, const c10::Device& device) { in searchsorted_scalar_tensor() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Bucketization.cu | 186 const Tensor& scalar_tensor = searchsorted_scalar_tensor(self, sorted_sequence.device()); in searchsorted_out_cuda()
211 const Tensor& scalar_tensor = searchsorted_scalar_tensor(self, sorted_sequence.device()); in searchsorted_cuda()
230 …return bucketize_cuda(searchsorted_scalar_tensor(self, boundaries.device()), boundaries, out_int32… in bucketize_cuda()
|