| 53 | torch::nn::Sequential ASPPSeparableConv(int in_channels, int out_channels, int dilation); |
| 54 | |
| 55 | class ASPPPoolingImpl : public torch::nn::Module { |
| 56 | public: |
| 57 | torch::nn::Sequential seq{nullptr}; |
| 58 | ASPPPoolingImpl(int in_channels, int out_channels); |
| 59 | torch::Tensor forward(torch::Tensor x); |
| 60 | |
| 61 | }; TORCH_MODULE(ASPPPooling); |
| 62 | |
| 63 | class ASPPImpl : public torch::nn::Module { |
| 64 | public: |
nothing calls this directly
no outgoing calls
no test coverage detected