Searched refs:max_pool2d_backward_fn (Results 1 – 1 of 1) sorted by relevance
15 using max_pool2d_backward_fn = void(*)(const Tensor& grad_input, const Tensor& grad_output, const T… variable18 DECLARE_DISPATCH(max_pool2d_backward_fn, max_pool2d_backward_kernel);