Home
last modified time | relevance | path

Searched refs:conv2D (Results 1 – 25 of 26) sorted by relevance

12

/dports/misc/mnn/MNN-1.2.0/tools/train/source/grad/
H A DConvGrad.cpp36 auto conv2D = new Convolution2DT; in onGrad() local
40 auto padMode = conv2D->common->padMode; in onGrad()
41 if ((conv2D->common->strideX > 1 || conv2D->common->strideY > 1)) { in onGrad()
51 auto kW = conv2D->common->kernelX; in onGrad()
52 auto kH = conv2D->common->kernelY; in onGrad()
53 auto sW = conv2D->common->strideX; in onGrad()
54 auto sH = conv2D->common->strideY; in onGrad()
55 auto dW = conv2D->common->dilateX; in onGrad()
77 conv2D->common->pads = padding; in onGrad()
98 newOp->main.value = conv2D; in onGrad()
[all …]
/dports/misc/mnn/MNN-1.2.0/express/
H A DNeuralNetWorkOp.cpp192 conv2D->common->relu6 = relu6; in _Conv()
193 conv2D->common->relu = relu; in _Conv()
197 conv2D->bias = std::move(bias); in _Conv()
227 conv2D->common->relu6 = relu6; in _Conv()
228 conv2D->common->relu = relu; in _Conv()
233 conv2D->weight.clear(); in _Conv()
260 std::fill(conv2D->weight.begin(), conv2D->weight.end(), weight); in _Conv()
262 std::fill(conv2D->bias.begin(), conv2D->bias.end(), bias); in _Conv()
330 conv2D->common->relu6 = relu6; in _Deconv()
331 conv2D->common->relu = relu; in _Deconv()
[all …]
/dports/misc/mnn/MNN-1.2.0/tools/converter/source/optimizer/postconvert/
H A DMergeScaleToConvolution.cpp29 auto conv2D = convolutionOp->main.AsConvolution2D(); in merge2Convolution() local
30 int outputCount = conv2D->common->outputCount; in merge2Convolution()
32 conv2D->bias[i] = conv2D->bias[i] * alpha[i] + bias[i]; in merge2Convolution()
35 if (nullptr != conv2D->quanParameter.get()) { in merge2Convolution()
37 conv2D->quanParameter->alpha[i] *= alpha[i]; in merge2Convolution()
40 int weightPartSize = conv2D->weight.size() / outputCount; in merge2Convolution()
43 … conv2D->weight.size() / outputCount / conv2D->common->kernelX / conv2D->common->kernelY; in merge2Convolution()
45 … auto dstPos = i * outputCount * conv2D->common->kernelY * conv2D->common->kernelX; in merge2Convolution()
47 … auto dstPosJ = dstPos + j * conv2D->common->kernelY * conv2D->common->kernelX; in merge2Convolution()
49 … for (int k = 0; k < conv2D->common->kernelY * conv2D->common->kernelX; ++k) { in merge2Convolution()
[all …]
H A DMergeBNToConvolution.cpp41 auto conv2D = convolutionOp->main.AsConvolution2D(); in merge2Convolution() local
42 int outputCount = conv2D->common->outputCount; in merge2Convolution()
44 conv2D->bias[i] = conv2D->bias[i] * alpha[i] + bias[i]; in merge2Convolution()
47 if (nullptr != conv2D->quanParameter.get()) { in merge2Convolution()
49 conv2D->quanParameter->alpha[i] *= alpha[i]; in merge2Convolution()
52 int weightPartSize = conv2D->weight.size() / outputCount; in merge2Convolution()
55 … conv2D->weight.size() / outputCount / conv2D->common->kernelX / conv2D->common->kernelY; in merge2Convolution()
57 … auto dstPos = i * outputCount * conv2D->common->kernelY * conv2D->common->kernelX; in merge2Convolution()
59 … auto dstPosJ = dstPos + j * conv2D->common->kernelY * conv2D->common->kernelX; in merge2Convolution()
61 … for (int k = 0; k < conv2D->common->kernelY * conv2D->common->kernelX; ++k) { in merge2Convolution()
[all …]
H A DTransformGroupConvolution.cpp18 auto conv2D = op->main.AsConvolution2D(); in onExecute() local
20 auto& common = conv2D->common; in onExecute()
23 if (nullptr != conv2D->quanParameter.get()) { in onExecute()
24 auto& quanParam = conv2D->quanParameter; in onExecute()
67 auto conv2D = op->main.AsConvolution2D(); in onExecute() local
68 auto& common = conv2D->common; in onExecute()
115 int partWeightSize = conv2D->weight.size() / common->group; in onExecute()
116 int partBiasSize = conv2D->bias.size() / common->group; in onExecute()
150 newConvolutionT->weight.push_back(conv2D->weight[startWeight + v]); in onExecute()
153 newConvolutionT->bias.push_back(conv2D->bias[startBias + v]); in onExecute()
/dports/misc/mnn/MNN-1.2.0/test/op/
H A DConvolutionTest.cpp119 conv2D->common->group = group;
193 conv2D->common->group = group;
200 conv2D->common->relu6 = relu6;
201 conv2D->common->relu = relu;
242 conv2D->weight = std::move(weight);
244 conv2D->bias = std::move(bias);
262 conv2D->common->group = group;
270 std::fill(conv2D->weight.begin(), conv2D->weight.end(), weight);
271 conv2D->bias.resize(channel[1]);
272 std::fill(conv2D->bias.begin(), conv2D->bias.end(), bias);
[all …]
/dports/misc/mnn/MNN-1.2.0/tools/cpp/
H A DConvertToFullQuant.hpp114 auto conv2D = op->main.AsConvolution2D(); in ConvertOp() local
117 if (conv2D->symmetricQuan && (!conv2D->symmetricQuan->weight.empty())) { in ConvertOp()
119 if (conv2D->quanParameter && conv2D->quanParameter->buffer.empty()) { in ConvertOp()
120 auto aMin = conv2D->quanParameter->aMin; in ConvertOp()
121 auto scaleIn = conv2D->quanParameter->scaleIn; in ConvertOp()
122 auto scaleOut = conv2D->quanParameter->scaleOut; in ConvertOp()
123 auto weightScale = conv2D->quanParameter->alpha; in ConvertOp()
127 const int kn = conv2D->common->outputCount; in ConvertOp()
137 conv2D->quanParameter->scaleIn = scaleIn; in ConvertOp()
138 conv2D->quanParameter->scaleOut = scaleOut; in ConvertOp()
[all …]
/dports/misc/mnn/MNN-1.2.0/source/backend/tensorrt/execution/
H A DTRTConvolution.cpp26 auto conv2D = mOp->main_as_Convolution2D(); in onEncode() local
27 auto conv2DCommon = conv2D->common(); in onEncode()
42 if (nullptr != conv2D->weight()) { in onEncode()
43 srcCount = conv2D->weight()->size() / (outputCount * kernelX * kernelY); in onEncode()
44 source = conv2D->weight()->data(); in onEncode()
45 weightSize = conv2D->weight()->size(); in onEncode()
47 srcCount = conv2D->common()->inputCount(); in onEncode()
60 …TRTWeight bias{nvinfer1::DataType::kFLOAT, static_cast<void *>(const_cast<float *>(conv2D->bias()-… in onEncode()
61 static_cast<size_t>(conv2D->bias()->size())}; in onEncode()
H A DTRTDepthwiseConvolution.cpp26 auto conv2D = mOp->main_as_Convolution2D(); in onEncode() local
27 auto conv2DCommon = conv2D->common(); in onEncode()
43 if (nullptr != conv2D->weight()) { in onEncode()
44 source = conv2D->weight()->data(); in onEncode()
45 weightSize = conv2D->weight()->size(); in onEncode()
52 …TRTWeight bias{nvinfer1::DataType::kFLOAT, static_cast<void *>(const_cast<float *>(conv2D->bias()-… in onEncode()
53 static_cast<size_t>(conv2D->bias()->size())}; in onEncode()
H A DTRTDeconvolution.cpp28 auto conv2D = mOp->main_as_Convolution2D(); in onEncode() local
29 auto conv2DCommon = conv2D->common(); in onEncode()
38 ConvolutionCommon::getConvParameters(&quanCommon, conv2D, &source, &weightSize); in onEncode()
46 …TRTWeight bias{nvinfer1::DataType::kFLOAT, static_cast<void *>(const_cast<float *>(conv2D->bias()-… in onEncode()
47 static_cast<size_t>(conv2D->bias()->size())}; in onEncode()
H A DTRTDepthwiseDeconvolution.cpp28 auto conv2D = mOp->main_as_Convolution2D(); in onEncode() local
29 auto conv2DCommon = conv2D->common(); in onEncode()
38 ConvolutionCommon::getConvParameters(&quanCommon, conv2D, &source, &weightSize); in onEncode()
46 …TRTWeight bias{nvinfer1::DataType::kFLOAT, static_cast<void *>(const_cast<float *>(conv2D->bias()-… in onEncode()
47 static_cast<size_t>(conv2D->bias()->size())}; in onEncode()
/dports/misc/mnn/MNN-1.2.0/source/backend/hiai/execution/
H A DNPUConvolution.cpp27 auto conv2D = mOp->main_as_Convolution2D(); in onResize() local
28 auto conv2DCommon = conv2D->common(); in onResize()
38 if (nullptr != conv2D->quanParameter()) { in onResize()
39 quanCommon = ConvolutionCommon::load(conv2D->quanParameter(), true); in onResize()
52 weightSize = conv2D->weight()->size(); in onResize()
53 filterDataPtr = conv2D->weight()->data(); in onResize()
80 filter->SetData((uint8_t *)conv2D->bias()->data(), conv2D->bias()->size() * sizeof(float)); in onResize()
H A DNPUConvolutionDepthwise.cpp27 auto conv2D = mOp->main_as_Convolution2D(); in onResize() local
28 auto conv2DCommon = conv2D->common(); in onResize()
38 if (nullptr != conv2D->quanParameter()) { in onResize()
39 quanCommon = ConvolutionCommon::load(conv2D->quanParameter(), true); in onResize()
52 weightSize = conv2D->weight()->size(); in onResize()
53 filterDataPtr = conv2D->weight()->data(); in onResize()
79 filter->SetData((uint8_t *)conv2D->bias()->data(), conv2D->bias()->size() * sizeof(float)); in onResize()
H A DNPUDeconvolution.cpp27 auto conv2D = mOp->main_as_Convolution2D(); in onResize() local
28 auto conv2DCommon = conv2D->common(); in onResize()
38 weightSize = conv2D->weight()->size(); in onResize()
39 filterDataPtr = conv2D->weight()->data(); in onResize()
66 filter->SetData((uint8_t *)conv2D->bias()->data(), conv2D->bias()->size() * sizeof(float)); in onResize()
H A DNPUConvolutionDepthwiseInt8.cpp27 auto conv2D = mOp->main_as_Convolution2D(); in onResize() local
28 auto conv2DCommon = conv2D->common(); in onResize()
29 auto quantizedParams = conv2D->symmetricQuan(); in onResize()
H A DNPUConvolutionInt8.cpp27 auto conv2D = mOp->main_as_Convolution2D(); in onResize() local
28 auto conv2DCommon = conv2D->common(); in onResize()
29 auto quantizedParams = conv2D->symmetricQuan(); in onResize()
/dports/misc/mnn/MNN-1.2.0/source/backend/coreml/execution/
H A DCoreMLConvolution.cpp29 auto conv2D = mOp->main_as_Convolution2D(); in loadWeightBias() local
30 if (nullptr != conv2D->quanParameter()) { in loadWeightBias()
31 quanCommon = ConvolutionCommon::load(conv2D->quanParameter(), true); in loadWeightBias()
42 weightSize = conv2D->weight()->size(); in loadWeightBias()
43 weightPtr = conv2D->weight()->data(); in loadWeightBias()
45 biasSize = conv2D->bias()->size(); in loadWeightBias()
46 biasPtr = conv2D->bias()->data(); in loadWeightBias()
100 auto conv2D = mOp->main_as_Convolution2D(); in onResize() local
101 auto common = conv2D->common(); in onResize()
/dports/misc/mnn/MNN-1.2.0/tools/train/source/transformer/
H A DOpConverter.cpp43 auto conv2D = op->main.AsConvolution2D(); in convert() local
44 auto conv2DCommon = conv2D->common.get(); in convert()
54 … auto srcCount = (int)conv2D->weight.size() * conv2DCommon->group / conv2DCommon->outputCount / in convert()
66 biasValue = _Const((const void*)conv2D->bias.data(), {(int)conv2D->bias.size()}, NCHW); in convert()
/dports/misc/mnn/MNN-1.2.0/tools/train/source/nn/
H A DNN.cpp428 option.kernelSize = {conv2D->common()->kernelX(), conv2D->common()->kernelY()}; in ExtractConvolution()
429 option.stride = {conv2D->common()->strideX(), conv2D->common()->strideY()}; in ExtractConvolution()
430 if (nullptr != conv2D->common()->pads()) { in ExtractConvolution()
436 option.pads = {conv2D->common()->padX(), conv2D->common()->padY()}; in ExtractConvolution()
438 switch (conv2D->common()->padMode()) { in ExtractConvolution()
451 option.dilate = {conv2D->common()->dilateX(), conv2D->common()->dilateY()}; in ExtractConvolution()
463 if (nullptr == conv2D->weight()) { in ExtractConvolution()
471 …inputCount = weightCount / conv2D->common()->kernelX() / conv2D->common()->kernelY() / conv2D->com… in ExtractConvolution()
491 if (conv2D->weight() == nullptr || conv2D->bias() == nullptr) { in ExtractConvolution()
501 if (conv2D->common()->relu()) { in ExtractConvolution()
[all …]
/dports/lang/halide/Halide-release_2019_08_27-2654-g664dc4993/apps/resnet_50/
H A DResnet50Generator.cpp165 conv1 = conv2D(input_t, conv1_ws, conv1_weights, "conv1"); in generate()
179 … br1_conv[br1_i] = conv2D(br2a_input, br1_ws[br1_i], br1_conv_weights[br1_i], "br1_conv"); in generate()
190 …br2a_conv[block_id] = conv2D(br2a_input, br2a_ws[block_id], weights, "block" + std::to_string(bloc… in generate()
197 …br2b_conv[block_id] = conv2D(br2a_relu[block_id], br2b_ws[block_id], weights, "block" + std::to_st… in generate()
204 …br2c_conv[block_id] = conv2D(br2b_relu[block_id], br2c_ws[block_id], weights, "block" + std::to_st… in generate()
253 …Tensor conv2D(const Tensor &input, const WeightShape &weight_shape, const Func &weights, const std… in conv2D() function in __anone42134760111::Resnet50Generator
/dports/misc/mnn/MNN-1.2.0/source/backend/cpu/
H A DCPUDeconvolution.cpp42 auto conv2D = convOp->main_as_Convolution2D(); in CPUDeconvolutionCommon() local
53 … ::memcpy(mBias->host<float>(), conv2D->bias()->data(), conv2D->bias()->size() * sizeof(float)); in CPUDeconvolutionCommon()
55 core->MNNFp32ToLowp(conv2D->bias()->data(), mBias->host<int16_t>(), conv2D->bias()->size()); in CPUDeconvolutionCommon()
/dports/misc/mnn/MNN-1.2.0/source/backend/cpu/compute/
H A DDeconvolutionWithStride.cpp167 auto conv2D = convOp->main_as_Convolution2D(); in DeconvolutionWithStride() local
168 MNN_ASSERT(nullptr != conv2D->bias()); in DeconvolutionWithStride()
169 auto common = conv2D->common(); in DeconvolutionWithStride()
180 ConvolutionCommon::getConvParameters(&quanCommon, conv2D, &tempWeight, &tempWeightSize); in DeconvolutionWithStride()
260 auto conv2D = convOp->main_as_Convolution2D(); in _extract() local
261 MNN_ASSERT(nullptr != conv2D->bias()); in _extract()
262 auto common = conv2D->common(); in _extract()
273 ConvolutionCommon::getConvParameters(&quanCommon, conv2D, &tempWeight, &tempWeightSize); in _extract()
/dports/misc/mnn/MNN-1.2.0/source/backend/opencl/execution/image/
H A DConvExecution.cpp440 auto conv2D = op->main_as_Convolution2D(); in onCreate() local
441 if (ConvWinograd::valid(conv2D->common(), inputs[0])) { in onCreate()
442 return new ConvWinograd(conv2D, backend); in onCreate()
/dports/misc/mnn/MNN-1.2.0/source/backend/opencl/execution/buffer/
H A DConvBufExecution.cpp649 auto conv2D = op->main_as_Convolution2D(); in onCreate() local
650 if (ConvBufWinograd::valid(conv2D->common(), inputs[0])) { in onCreate()
651 return new ConvBufWinograd(conv2D, backend); in onCreate()
/dports/math/PDL/PDL-2.019/
H A DChanges2458 conv1d() algorithm adjusted to match conv2D() and convolutionND().

12