Searched refs:inplaceOp (Results 1 – 5 of 5) sorted by relevance
| /dports/misc/mnn/MNN-1.2.0/tools/converter/source/optimizer/postconvert/ |
| H A D | MergeRelu6ToConvolution.cpp | 16 bool merge2Convolution(const MNN::OpT* inplaceOp, MNN::OpT* convolutionOp) const { in merge2Convolution() argument
17 if (inplaceOp->type == MNN::OpType_ReLU6) { in merge2Convolution()
18 if (nullptr == inplaceOp->main.AsRelu6()) { in merge2Convolution()
22 … if (inplaceOp->main.AsRelu6()->maxValue == 6.0f && inplaceOp->main.AsRelu6()->minValue == 0.0f) { in merge2Convolution()
30 bool merge2Convolution3D(const MNN::OpT* inplaceOp, MNN::OpT* convolutionOp) const { in merge2Convolution3D() argument
31 if (inplaceOp->type == MNN::OpType_ReLU6) { in merge2Convolution3D()
|
| H A D | MergeReluToConvolution.cpp | 16 bool merge2Convolution(const MNN::OpT* inplaceOp, MNN::OpT* convolutionOp) const { in merge2Convolution() argument
17 if (inplaceOp->type == MNN::OpType_ReLU && inplaceOp->main.AsRelu()->slope == 0.0f) { in merge2Convolution()
24 bool merge2Convolution3D(const MNN::OpT* inplaceOp, MNN::OpT* convolutionOp) const { in merge2Convolution3D() argument
25 if (inplaceOp->type == MNN::OpType_ReLU && inplaceOp->main.AsRelu()->slope == 0.0f) { in merge2Convolution3D()
|
| H A D | MergeScaleToConvolution.cpp | 16 bool merge2Convolution(const MNN::OpT* inplaceOp, MNN::OpT* convolutionOp) const { in merge2Convolution() argument
22 if (inplaceOp->type == MNN::OpType_Scale) { in merge2Convolution()
26 bias = inplaceOp->main.AsScale()->biasData; in merge2Convolution()
27 alpha = inplaceOp->main.AsScale()->scaleData; in merge2Convolution()
68 bool merge2Convolution3D(const MNN::OpT* inplaceOp, MNN::OpT* convolutionOp) const { in merge2Convolution3D() argument
74 if (inplaceOp->type == MNN::OpType_Scale) { in merge2Convolution3D()
78 bias = inplaceOp->main.AsScale()->biasData; in merge2Convolution3D()
79 alpha = inplaceOp->main.AsScale()->scaleData; in merge2Convolution3D()
|
| H A D | MergeBNToConvolution.cpp | 16 bool merge2Convolution(const MNN::OpT* inplaceOp, MNN::OpT* convolutionOp) const { in merge2Convolution() argument
22 if (inplaceOp->type == MNN::OpType_BatchNorm) { in merge2Convolution()
26 auto l = inplaceOp->main.AsBatchNorm(); in merge2Convolution()
80 bool merge2Convolution3D(const MNN::OpT* inplaceOp, MNN::OpT* convolutionOp) const { in merge2Convolution3D() argument
86 if (inplaceOp->type == MNN::OpType_BatchNorm) { in merge2Convolution3D()
90 auto l = inplaceOp->main.AsBatchNorm(); in merge2Convolution3D()
|
| H A D | MergeToConvolution.hpp | 14 virtual bool merge2Convolution(const MNN::OpT* inplaceOp, MNN::OpT* convolutionOp) const = 0;
16 virtual bool merge2Convolution3D(const MNN::OpT* inplaceOp, MNN::OpT* convolutionOp) const = 0;
|