1 #pragma once 2 #include <ATen/native/DispatchStub.h> 3 4 namespace at { 5 class Tensor; 6 7 namespace native { 8 9 using max_unpooling_fn = void(*)(Tensor&, const Tensor&, const Tensor&); 10 11 DECLARE_DISPATCH(max_unpooling_fn, max_unpool2d_kernel); 12 DECLARE_DISPATCH(max_unpooling_fn, max_unpool3d_kernel); 13 14 }} // at::native 15