Home
last modified time | relevance | path

Searched refs:ConvolutionCommon (Results 1 – 25 of 55) sorted by relevance

123

/dports/misc/mnn/MNN-1.2.0/tools/converter/source/caffe/
H A DConvolution.cpp14 class ConvolutionCommon : public OpConverter { class
128 ConvolutionCommon() { in ConvolutionCommon() function in ConvolutionCommon
130 virtual ~ConvolutionCommon() { in ~ConvolutionCommon()
140 class Convolution : public ConvolutionCommon {
143 ConvolutionCommon::run(dstOp, parameters, weight); in run()
185 class ConvolutionDepthwise : public ConvolutionCommon {
188 ConvolutionCommon::run(dstOp, parameters, weight); in run()
/dports/misc/mnn/MNN-1.2.0/source/backend/cpu/compute/
H A DConvolutionIntFactory.cpp15 … Backend *backend, const ConvolutionCommon::Int8Common *common, const float *bias, in createUnit()
22 const ConvolutionCommon::Int8Common *common) { in create()
38 auto subCommon = std::make_shared<ConvolutionCommon::Int8Common>(); in create()
H A DConvolutionIntFactory.hpp21 const ConvolutionCommon::Int8Common* common);
24 … const ConvolutionCommon::Int8Common* common, const float* bias, size_t biasSize);
H A DConvolutionInt8Executor.hpp22 … const ConvolutionCommon::Int8Common *common, const float *bias, size_t biasSize);
36 ConvolutionCommon::Im2ColParameter mIm2ColParamter;
H A DInt8FunctionsOpt.h63 … const ConvolutionCommon::Im2ColParameter* im2colParameter, size_t xIndexStart,
65 …Im2ColFunc(*chooseIm2Col)(const ConvolutionCommon::Im2ColParameter* im2colParam, size_t inputChann…
H A DInt8FunctionsOpt.cpp146 … const MNN::ConvolutionCommon::Im2ColParameter* im2colParameter, size_t xIndexStart, in _fastIm2Col()
172 … const MNN::ConvolutionCommon::Im2ColParameter* im2colParameter, size_t xIndexStart, in _im2colCommonZ1()
218 … const MNN::ConvolutionCommon::Im2ColParameter* im2colParameter, size_t xIndexStart, in _im2colCommon()
269 static MNN::CoreInt8Functions::Im2ColFunc chooseIm2Col(const MNN::ConvolutionCommon::Im2ColParamete… in chooseIm2Col()
305 … const MNN::ConvolutionCommon::Im2ColParameter* im2colParameter, size_t xIndexStart, in _im2colCommonSdot()
354 … const MNN::ConvolutionCommon::Im2ColParameter* im2colParameter, size_t xIndexStart, in _fastIm2ColSdot()
372 static MNN::CoreInt8Functions::Im2ColFunc chooseIm2ColSdot(const MNN::ConvolutionCommon::Im2ColPara… in chooseIm2ColSdot()
H A DConvolutionFloatFactory.cpp85 std::shared_ptr<ConvolutionCommon::Int8Common> quanCommon; in create()
87 quanCommon = ConvolutionCommon::load(conv2d->quanParameter()); in create()
/dports/misc/mnn/MNN-1.2.0/source/backend/tensorrt/execution/
H A DTRTDeconvolution.cpp37 std::shared_ptr<ConvolutionCommon::Int8Common> quanCommon; in onEncode()
38 ConvolutionCommon::getConvParameters(&quanCommon, conv2D, &source, &weightSize); in onEncode()
54 auto pads = ConvolutionCommon::convolutionPad(mInputs[0], mOutputs[0], conv2DCommon); in onEncode()
H A DTRTDepthwiseDeconvolution.cpp37 std::shared_ptr<ConvolutionCommon::Int8Common> quanCommon; in onEncode()
38 ConvolutionCommon::getConvParameters(&quanCommon, conv2D, &source, &weightSize); in onEncode()
54 auto pads = ConvolutionCommon::convolutionPad(mInputs[0], mOutputs[0], conv2DCommon); in onEncode()
H A DTRTDepthwiseConvolution.cpp37 std::shared_ptr<ConvolutionCommon::Int8Common> quanWeight; in onEncode()
39 quanWeight = ConvolutionCommon::load(mOp->main_as_Convolution2D()->quanParameter(), true); in onEncode()
60 auto pads = ConvolutionCommon::convolutionPad(mInputs[0], mOutputs[0], conv2DCommon); in onEncode()
H A DTRTConvolution.cpp35 std::shared_ptr<ConvolutionCommon::Int8Common> quanWeight; in onEncode()
37 quanWeight = ConvolutionCommon::load(mOp->main_as_Convolution2D()->quanParameter(), true); in onEncode()
82 auto pads = ConvolutionCommon::convolutionPad(mInputs[0], mOutputs[0], conv2DCommon); in onEncode()
/dports/misc/mnn/MNN-1.2.0/test/core/
H A DIDSTTest.cpp31 std::shared_ptr<ConvolutionCommon::Int8Common> common = ConvolutionCommon::load(idstQuant); in run()
/dports/misc/mnn/MNN-1.2.0/source/core/
H A DConvolutionCommon.hpp15 class MNN_PUBLIC ConvolutionCommon : public Execution { class
24 …static void getConvParameters(std::shared_ptr<ConvolutionCommon::Int8Common> *quanCommon, const MN…
H A DConvolutionCommon.cpp368 std::shared_ptr<ConvolutionCommon::Int8Common> ConvolutionCommon::load(const IDSTQuan *quan, bool f… in load()
466 void ConvolutionCommon::getConvParameters(std::shared_ptr<Int8Common> *quanCommon, const MNN::Convo… in getConvParameters()
480 bool ConvolutionCommon::getConvInt8Parameters(const MNN::Convolution2D* conv2d, std::shared_ptr<Int… in getConvInt8Parameters()
486 quanCommon = ConvolutionCommon::load(conv2d->quanParameter(), false, true); in getConvInt8Parameters()
543 std::pair<int, int> ConvolutionCommon::convolutionPad(const Tensor *input, const Tensor *output, in convolutionPad()
564 std::tuple<int, int, int, int> ConvolutionCommon::convolutionPadFull(const Tensor* input, const Ten… in convolutionPadFull()
585 std::pair<int, int> ConvolutionCommon::convolutionTransposePad(const Tensor *input, const Tensor *o… in convolutionTransposePad()
/dports/misc/mnn/MNN-1.2.0/source/backend/metal/
H A DMetalConvolutionCommon.mm91 std::shared_ptr<ConvolutionCommon::Int8Common> qnt = NULL;
93 qnt = ConvolutionCommon::load(conv->quanParameter(), true);
105 id<MTLBuffer> MetalConvolutionCommon::weightForConv(const Convolution2D *conv, ConvolutionCommon::I…
H A DMetalDeconvolution.mm10 #import "core/ConvolutionCommon.hpp"
89 ConvolutionCommon::Int8Common *qnt) {
137 std::shared_ptr<ConvolutionCommon::Int8Common> qnt = NULL;
139 qnt = ConvolutionCommon::load(deconv->quanParameter(), true);
/dports/misc/mnn/MNN-1.2.0/source/backend/tensorrt/backend/
H A DTRTBackend.hpp112 void pushCache(std::shared_ptr<ConvolutionCommon::Int8Common> cache) const { in pushCache()
120 …mutable std::vector<std::shared_ptr<ConvolutionCommon::Int8Common>> mCache; // should be deleted a…
/dports/misc/mnn/MNN-1.2.0/source/backend/vulkan/execution/
H A DVulkanDeconvolution.cpp36 std::shared_ptr<ConvolutionCommon::Int8Common> quanCommon; in VulkanDeconvolution()
37 ConvolutionCommon::getConvParameters(&quanCommon, conv, &filterDataPtr, &tempWeightSize); in VulkanDeconvolution()
95 auto pad = ConvolutionCommon::convolutionTransposePad(src, dst, common); in writeConvolutionConst()
H A DVulkanDeconvolutionDepthwise.cpp43 std::shared_ptr<ConvolutionCommon::Int8Common> quanCommon; in VulkanDeconvolutionDepthwise()
44 ConvolutionCommon::getConvParameters(&quanCommon, conv, &tempWeight, &tempWeightSize); in VulkanDeconvolutionDepthwise()
/dports/misc/mnn/MNN-1.2.0/source/backend/hiai/execution/
H A DNPUConvolution.cpp37 std::shared_ptr<MNN::ConvolutionCommon::Int8Common> quanCommon; in onResize()
39 quanCommon = ConvolutionCommon::load(conv2D->quanParameter(), true); in onResize()
H A DNPUConvolutionDepthwise.cpp37 std::shared_ptr<MNN::ConvolutionCommon::Int8Common> quanCommon; in onResize()
39 quanCommon = ConvolutionCommon::load(conv2D->quanParameter(), true); in onResize()
/dports/misc/mnn/MNN-1.2.0/source/backend/coreml/execution/
H A DCoreMLConvolution.hpp27 std::shared_ptr<ConvolutionCommon::Int8Common> quanCommon;
/dports/misc/mnn/MNN-1.2.0/source/backend/cpu/
H A DOneDNNConvInt8.cpp69 std::shared_ptr<ConvolutionCommon::Int8Common> quanCommon; in create()
71 quanCommon = ConvolutionCommon::load(convParam->quanParameter(), false); in create()
110 auto pads = ConvolutionCommon::convolutionPadFull(inputs[0], outputs[0], mCommon); in onResize()
H A DCPUTFQuantizedConv2D.hpp51 ConvolutionCommon::Im2ColParameter *mIm2ColParamter;
H A DCPUConvolution.cpp154 std::shared_ptr<ConvolutionCommon::Int8Common> quanCommon; in makeResourceInt8()
155 … if (!ConvolutionCommon::getConvInt8Parameters(convParam, quanCommon, weightSrc, scalePtr, biasPtr, in makeResourceInt8()
252 auto pad = ConvolutionCommon::convolutionPad(input, output, mCommon); in onResize()

123