Home
last modified time | relevance | path

Searched refs:cpu_adaptive_max_pool2d (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DAdaptiveMaxPoolKernel.cpp18 void cpu_adaptive_max_pool2d( in cpu_adaptive_max_pool2d() function
445cpu_adaptive_max_pool2d<scalar_t, /*accscalar_t*/param_t>(output, indices, input, output_size); in adaptive_max_pool2d_kernel_impl()