Searched refs:cpu_max_unpool (Results 1 – 1 of 1) sorted by relevance
17 void cpu_max_unpool( in cpu_max_unpool() function243 cpu_max_unpool<scalar_t, /*is_3d*/false>(output, input, indices); in max_unpool2d_kernel_impl()263 cpu_max_unpool<scalar_t, /*is_3d*/true>(output, input, indices); in max_unpool3d_kernel_impl()