Searched refs:upsample_nearest3d_out_cuda_template (Results 1 – 1 of 1) sorted by relevance
147 static void upsample_nearest3d_out_cuda_template( in upsample_nearest3d_out_cuda_template() function302 upsample_nearest3d_out_cuda_template<nearest_neighbor_compute_source_index>( in TORCH_IMPL_FUNC()313 …upsample_nearest3d_out_cuda_template<nearest_neighbor_exact_compute_source_index>(output, input, o… in TORCH_IMPL_FUNC()