Searched refs:cpu_max_pool_channels_last (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | MaxPoolKernel.cpp | 356 void cpu_max_pool_channels_last( in cpu_max_pool_channels_last() function
616 …cpu_max_pool_channels_last<scalar_t, false>(output, indices, input, {kW, kH}, {dW, dH}, {padW, pad… in max_pool2d_kernel_impl()
649 cpu_max_pool_channels_last<scalar_t, /*is 3d*/true>(output, indices, input_cl_check, in max_pool3d_kernel_impl()
667 cpu_max_pool_channels_last<scalar_t, true>(output, indices, input, in max_pool3d_kernel_impl()
|