Home
last modified time | relevance | path

Searched refs:nearest_neighbor_compute_source_index (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DUpSample.h311 inline int64_t nearest_neighbor_compute_source_index( in nearest_neighbor_compute_source_index() function
351 return nearest_neighbor_compute_source_index(scale, output_index, input_size); in nearest_idx()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A DUpSampleNearest2d.cpp210 …return _upsample_nearest2d_quantized_cpu<nearest_neighbor_compute_source_index>(input, osize, scal… in upsample_nearest2d_quantized_cpu()
H A DUpSampleNearest3d.cpp223 return _upsample_nearest3d_quantized_cpu<nearest_neighbor_compute_source_index>( in upsample_nearest3d_quantized_cpu()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DUpSampleNearest1d.cu206 upsample_nearest1d_out_cuda_template<nearest_neighbor_compute_source_index>( in TORCH_IMPL_FUNC()
H A DUpSample.cuh133 __device__ __forceinline__ int nearest_neighbor_compute_source_index( in nearest_neighbor_compute_source_index() function
H A DUpSampleNearest3d.cu302 upsample_nearest3d_out_cuda_template<nearest_neighbor_compute_source_index>( in TORCH_IMPL_FUNC()
H A DUpSampleNearest2d.cu452 upsample_nearest2d_out_cuda_template<nearest_neighbor_compute_source_index>( in TORCH_IMPL_FUNC()