Searched refs:cpu_upsample_nearest_backward (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | UpSampleMoreKernel.cpp | 92 void cpu_upsample_nearest_backward( in cpu_upsample_nearest_backward() function
342 …cpu_upsample_nearest_backward<scalar_t, scale_t, nearest_idx>(grad_input, grad_output, {scales_w}); in upsample_nearest1d_backward_kernel_impl()
351 …cpu_upsample_nearest_backward<scalar_t, scale_t, nearest_exact_idx>(grad_input, grad_output, {scal… in _upsample_nearest_exact1d_backward_kernel_impl()
366 …cpu_upsample_nearest_backward<scalar_t, scale_t, nearest_idx>(grad_input, grad_output, {scales_h, … in upsample_nearest2d_backward_kernel_impl()
382 …cpu_upsample_nearest_backward<scalar_t, scale_t, nearest_exact_idx>(grad_input, grad_output, {scal… in _upsample_nearest_exact2d_backward_kernel_impl()
399 …cpu_upsample_nearest_backward<scalar_t, scale_t, nearest_idx>(grad_input, grad_output, {scales_d, … in upsample_nearest3d_backward_kernel_impl()
416 …cpu_upsample_nearest_backward<scalar_t, scale_t, nearest_exact_idx>(grad_input, grad_output, {scal… in _upsample_nearest_exact3d_backward_kernel_impl()
|