MNN/source/backend/opencl/execution/buffer/DeconvBufExecution.hpp

36 lines
1.0 KiB
C++
Raw Normal View History

//
2023-07-18 09:36:26 +08:00
// DeconvBufExecution.hpp
// MNN
//
// Created by MNN on 2021/04/09.
// Copyright © 2018, Alibaba Group Holding Limited
//
2023-07-18 09:36:26 +08:00
#ifndef MNN_OPENCL_BUFFER_CLOSED
#ifndef DeconvBufExecution_hpp
#define DeconvBufExecution_hpp
#include "ConvBufExecution.hpp"
namespace MNN {
namespace OpenCL {
2024-04-19 11:58:21 +08:00
class DeconvBufExecution : public ConvBufCommonExecution, public CommonExecution {
public:
DeconvBufExecution(const std::vector<Tensor *> &inputs, const MNN::Op *op, Backend *backend);
2024-04-19 11:58:21 +08:00
DeconvBufExecution(std::shared_ptr<ConvBufResource> resource, const MNN::Op* op, Backend* backend);
virtual ~DeconvBufExecution();
2024-04-19 11:58:21 +08:00
virtual ErrorCode onEncode(const std::vector<Tensor *> &inputs, const std::vector<Tensor *> &outputs) override;
virtual bool onClone(Backend* bn, const Op* op, Execution** dst) override;
private:
std::vector<uint32_t> mLWS{0, 0, 0, 0};
std::vector<uint32_t> mGWS{0, 0, 0, 0};
};
} // namespace OpenCL
} // namespace MNN
#endif /* DeconvBufExecution_hpp */
#endif /* MNN_OPENCL_BUFFER_CLOSED */