Searched refs:cpu_padding_backward_channels_last (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | PaddingKernel.cpp | 396 void cpu_padding_backward_channels_last( in cpu_padding_backward_channels_last() function
549 cpu_padding_backward_channels_last<scalar_t, ReflectionPad>(grad_input, grad_output, param); in reflection_pad2d_backward_kernel_impl()
594 cpu_padding_backward_channels_last<scalar_t, ReflectionPad>(grad_input, grad_output, param); in reflection_pad3d_backward_kernel_impl()
657 … cpu_padding_backward_channels_last<scalar_t, ReplicationPad>(grad_input, grad_output, param); in replication_pad2d_backward_kernel_impl()
702 … cpu_padding_backward_channels_last<scalar_t, ReplicationPad>(grad_input, grad_output, param); in replication_pad3d_backward_kernel_impl()
|