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