Searched refs:cpu_max_pool_backward_channels_last (Results 1 – 1 of 1) sorted by relevance
538 void cpu_max_pool_backward_channels_last( in cpu_max_pool_backward_channels_last() function690 … cpu_max_pool_backward_channels_last<scalar_t, /*is 3d*/ false>(grad_input, grad_output, indices); in max_pool2d_backward_kernel_impl()717 …cpu_max_pool_backward_channels_last<scalar_t, /*is_3d*/ true>(grad_input, grad_output_cl_check, _i… in max_pool3d_backward_kernel_impl()732 … cpu_max_pool_backward_channels_last<scalar_t, /*is_3d*/ true>(grad_input, grad_output, indices); in max_pool3d_backward_kernel_impl()