Searched refs:nearest_neighbor_exact_bw_compute_source_index (Results 1 – 4 of 4) sorted by relevance
237 upsample_nearest1d_backward_out_cuda_template<nearest_neighbor_exact_bw_compute_source_index>( in TORCH_IMPL_FUNC()
173 __device__ __forceinline__ int nearest_neighbor_exact_bw_compute_source_index( in nearest_neighbor_exact_bw_compute_source_index() function
336 upsample_nearest3d_backward_out_cuda_template<nearest_neighbor_exact_bw_compute_source_index>( in TORCH_IMPL_FUNC()
484 upsample_nearest2d_backward_out_cuda_template<nearest_neighbor_exact_bw_compute_source_index>( in TORCH_IMPL_FUNC()