Searched refs:cpu_max_pool_backward (Results 1 – 1 of 1) sorted by relevance
473 void cpu_max_pool_backward( in cpu_max_pool_backward() function684 cpu_max_pool_backward<scalar_t, /*is 3d*/ false>(grad_input, grad_output, indices); in max_pool2d_backward_kernel_impl()726 cpu_max_pool_backward<scalar_t, /*is_3d*/ true>(grad_input, grad_output, indices); in max_pool3d_backward_kernel_impl()