Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DUpSampleNearest1d.cu226 upsample_nearest1d_backward_out_cuda_template<nearest_neighbor_bw_compute_source_index>( in TORCH_IMPL_FUNC()
H A DUpSample.cuh160 __device__ __forceinline__ int nearest_neighbor_bw_compute_source_index( in nearest_neighbor_bw_compute_source_index() function
H A DUpSampleNearest3d.cu324 upsample_nearest3d_backward_out_cuda_template<nearest_neighbor_bw_compute_source_index>( in TORCH_IMPL_FUNC()
H A DUpSampleNearest2d.cu473 upsample_nearest2d_backward_out_cuda_template<nearest_neighbor_bw_compute_source_index>( in TORCH_IMPL_FUNC()