// // InterpExecution.hpp // MNN // // Created by MNN on 2019/01/31. // Copyright © 2018, Alibaba Group Holding Limited // #ifndef InterpExecution_hpp #define InterpExecution_hpp #include "core/Execution.hpp" #include #include "backend/cuda/core/CUDABackend.hpp" namespace MNN { namespace CUDA { class InterpExecution : public Execution { public: InterpExecution(const Interp* interp, Backend *backend); virtual ~InterpExecution(); virtual ErrorCode onResize(const std::vector &inputs, const std::vector &outputs) override; virtual ErrorCode onExecute(const std::vector &inputs, const std::vector &outputs) override; private: CUDARuntime *mRuntime; float mWidthOffset; float mHeightOffset; int mResizeType; int mCount; int mBatch; int mChannel; int mInputHeight; int mInputWidth; int mOutputHeight; int mOutputWidth; float mScaleHeight; float mScaleWidth; }; } // namespace CUDA } // namespace MNN #endif /* InterpExecution_hpp */