Searched refs:cpu_upsample_nearest_backward_channels_last (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | UpSampleMoreKernel.cpp | 225 void cpu_upsample_nearest_backward_channels_last( in cpu_upsample_nearest_backward_channels_last() function
362 …cpu_upsample_nearest_backward_channels_last<scalar_t, scale_t, nearest_idx>(grad_input, grad_outpu… in upsample_nearest2d_backward_kernel_impl()
378 …cpu_upsample_nearest_backward_channels_last<scalar_t, scale_t, nearest_exact_idx>(grad_input, grad… in _upsample_nearest_exact2d_backward_kernel_impl()
395 …cpu_upsample_nearest_backward_channels_last<scalar_t, scale_t, nearest_idx>(grad_input, grad_outpu… in upsample_nearest3d_backward_kernel_impl()
412 …cpu_upsample_nearest_backward_channels_last<scalar_t, scale_t, nearest_exact_idx>(grad_input, grad… in _upsample_nearest_exact3d_backward_kernel_impl()
|