1 //
2 //  TRTDeconvolution.hpp
3 //  MNN
4 //
5 //  Created by MNN on 2019/09/11.
6 //  Copyright © 2018, Alibaba Group Holding Limited
7 //
8 
9 #ifndef MNN_TRTDeconvolution_HPP
10 #define MNN_TRTDeconvolution_HPP
11 
12 #include "TRTBackend.hpp"
13 #include "TRTCommonExecution.hpp"
14 
15 namespace MNN {
16 
17 class TRTDeconvolution : public TRTCommonExecution {
18 public:
19     TRTDeconvolution(Backend *b, const Op *op, const std::vector<Tensor *> &inputs, const std::vector<Tensor *> &outputs);
20     virtual ~TRTDeconvolution() = default;
21     virtual std::vector<ITensor *> onEncode(const std::vector<ITensor *> &inputs) override;
22 
23 private:
24     IActivationLayer *mActivationLayer{nullptr};
25 };
26 
27 } // namespace MNN
28 
29 #endif // MNN_TRTDeconvolution_HPP
30