| 25 | } |
| 26 | |
| 27 | ASPPPoolingImpl::ASPPPoolingImpl(int in_channels, int out_channels) { |
| 28 | seq = torch::nn::Sequential(torch::nn::AdaptiveAvgPool2d(torch::nn::AdaptiveAvgPool2dOptions(1)), |
| 29 | torch::nn::Conv2d(conv_options(in_channels, out_channels, 1, 1, 0, 1, false)), |
| 30 | torch::nn::BatchNorm2d(torch::nn::BatchNorm2dOptions(out_channels)), |
| 31 | torch::nn::ReLU()); |
| 32 | register_module("seq", seq); |
| 33 | } |
| 34 | |
| 35 | torch::Tensor ASPPPoolingImpl::forward(torch::Tensor x) { |
| 36 | auto residual(x.clone()); |
nothing calls this directly
no test coverage detected