| 61 | }; TORCH_MODULE(ASPPPooling); |
| 62 | |
| 63 | class ASPPImpl : public torch::nn::Module { |
| 64 | public: |
| 65 | ASPPImpl(int in_channels, int out_channels, std::vector<int> atrous_rates, bool separable = false); |
| 66 | torch::Tensor forward(torch::Tensor x); |
| 67 | private: |
| 68 | torch::nn::ModuleList modules{}; |
| 69 | ASPPPooling aspppooling{ nullptr }; |
| 70 | torch::nn::Sequential project{ nullptr }; |
| 71 | }; TORCH_MODULE(ASPP); |
| 72 | |
| 73 | class DeepLabV3DecoderImpl : public torch::nn::Module |
| 74 | { |
nothing calls this directly
no outgoing calls
no test coverage detected