Home
last modified time | relevance | path

Searched refs:searchsorted_maybe_trim_input_tensors (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DBucketizationUtils.h20 inline void searchsorted_maybe_trim_input_tensors( in searchsorted_maybe_trim_input_tensors() function
66 inline void searchsorted_maybe_trim_input_tensors( in searchsorted_maybe_trim_input_tensors() function
73 return searchsorted_maybe_trim_input_tensors( in searchsorted_maybe_trim_input_tensors()
H A DBucketization.cpp177 …searchsorted_maybe_trim_input_tensors(trimmed_input, trimmed_boundaries, trimmed_sorter, self, sor… in searchsorted_out_cpu()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DBucketization.cu164 …searchsorted_maybe_trim_input_tensors(trimmed_input, trimmed_boundaries, trimmed_sorter, self, sor… in searchsorted_out_cuda()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/operations/
H A DBucketization.mm284 searchsorted_maybe_trim_input_tensors(