1 // 2 // VulkanConvolutionImpl.hpp 3 // MNN 4 // 5 // Created by MNN on 2019/01/31. 6 // Copyright © 2018, Alibaba Group Holding Limited 7 // 8 9 #ifndef VulkanConvolutionImpl_hpp 10 #define VulkanConvolutionImpl_hpp 11 #include "VulkanBasicExecution.hpp" 12 namespace MNN { 13 class VulkanConvolutionImpl { 14 public: ~VulkanConvolutionImpl()15 virtual ~VulkanConvolutionImpl() { 16 } 17 18 static VulkanBasicExecution* create(VulkanBackend* backend, const Convolution2DCommon* convOption, 19 const std::vector<Tensor*>& input, const Tensor* output, const float* weightPtr, 20 const float* biasPtr, int ci, int co); 21 static int gImage2ColLocal; 22 }; 23 } // namespace MNN 24 #endif /* VulkanConvolutionImpl_hpp */ 25