1 #include <torch/nn/options/pooling.h> 2 3 namespace torch { 4 namespace nn { 5 6 template struct AvgPoolOptions<1>; 7 template struct AvgPoolOptions<2>; 8 template struct AvgPoolOptions<3>; 9 10 template struct MaxPoolOptions<1>; 11 template struct MaxPoolOptions<2>; 12 template struct MaxPoolOptions<3>; 13 14 template struct AdaptiveMaxPoolOptions<ExpandingArray<1>>; 15 template struct AdaptiveMaxPoolOptions<ExpandingArrayWithOptionalElem<2>>; 16 template struct AdaptiveMaxPoolOptions<ExpandingArrayWithOptionalElem<3>>; 17 18 template struct AdaptiveAvgPoolOptions<ExpandingArray<1>>; 19 template struct AdaptiveAvgPoolOptions<ExpandingArrayWithOptionalElem<2>>; 20 template struct AdaptiveAvgPoolOptions<ExpandingArrayWithOptionalElem<3>>; 21 22 template struct MaxUnpoolOptions<1>; 23 template struct MaxUnpoolOptions<2>; 24 template struct MaxUnpoolOptions<3>; 25 26 template struct LPPoolOptions<1>; 27 template struct LPPoolOptions<2>; 28 template struct LPPoolOptions<3>; 29 30 } // namespace nn 31 } // namespace torch 32