!7995 modify static check of onnx parser

Merge pull request !7995 from lyvette/master2
This commit is contained in:
mindspore-ci-bot 2020-10-30 09:19:41 +08:00 committed by Gitee
commit 00469ba584
44 changed files with 75 additions and 2 deletions

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxArgMaxParser : public OnnxNodeParser {
public:
OnnxArgMaxParser() : OnnxNodeParser("ArgMax") {}
~OnnxArgMaxParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,174 +25,203 @@ namespace lite {
class OnnxAddParser : public OnnxNodeParser {
public:
OnnxAddParser() : OnnxNodeParser("Add") {}
~OnnxAddParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};
class OnnxSubParser : public OnnxNodeParser {
public:
OnnxSubParser() : OnnxNodeParser("Sub") {}
~OnnxSubParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};
class OnnxMulParser : public OnnxNodeParser {
public:
OnnxMulParser() : OnnxNodeParser("Mul") {}
~OnnxMulParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};
class OnnxDivParser : public OnnxNodeParser {
public:
OnnxDivParser() : OnnxNodeParser("Div") {}
~OnnxDivParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};
class OnnxMeanParser : public OnnxNodeParser {
public:
OnnxMeanParser() : OnnxNodeParser("Mean") {}
~OnnxMeanParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};
class OnnxPowParser : public OnnxNodeParser {
public:
OnnxPowParser() : OnnxNodeParser("Power") {}
~OnnxPowParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};
class OnnxEqualParser : public OnnxNodeParser {
public:
OnnxEqualParser() : OnnxNodeParser("Equal") {}
~OnnxEqualParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};
class OnnxLessParser : public OnnxNodeParser {
public:
OnnxLessParser() : OnnxNodeParser("Less") {}
~OnnxLessParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};
class OnnxGreaterParser : public OnnxNodeParser {
public:
OnnxGreaterParser() : OnnxNodeParser("Greater") {}
~OnnxGreaterParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};
class OnnxMinParser : public OnnxNodeParser {
public:
OnnxMinParser() : OnnxNodeParser("Min") {}
~OnnxMinParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};
class OnnxEltwiseParser : public OnnxNodeParser {
public:
OnnxEltwiseParser() : OnnxNodeParser("Eltwise") {}
~OnnxEltwiseParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};
class OnnxFloorParser : public OnnxNodeParser {
public:
OnnxFloorParser() : OnnxNodeParser("Floor") {}
~OnnxFloorParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};
class OnnxAbsParser : public OnnxNodeParser {
public:
OnnxAbsParser() : OnnxNodeParser("Abs") {}
~OnnxAbsParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};
class OnnxNegParser : public OnnxNodeParser {
public:
OnnxNegParser() : OnnxNodeParser("Neg") {}
~OnnxNegParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};
class OnnxExpParser : public OnnxNodeParser {
public:
OnnxExpParser() : OnnxNodeParser("Exp") {}
~OnnxExpParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};
class OnnxCosParser : public OnnxNodeParser {
public:
OnnxCosParser() : OnnxNodeParser("Cos") {}
~OnnxCosParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};
class OnnxSinParser : public OnnxNodeParser {
public:
OnnxSinParser() : OnnxNodeParser("Sin") {}
~OnnxSinParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};
class OnnxSqrtParser : public OnnxNodeParser {
public:
OnnxSqrtParser() : OnnxNodeParser("Sqrt") {}
~OnnxSqrtParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};
class OnnxCeilParser : public OnnxNodeParser {
public:
OnnxCeilParser() : OnnxNodeParser("Ceil") {}
~OnnxCeilParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};
class OnnxLogParser : public OnnxNodeParser {
public:
OnnxLogParser() : OnnxNodeParser("Log") {}
~OnnxLogParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};
class OnnxTanParser : public OnnxNodeParser {
public:
OnnxTanParser() : OnnxNodeParser("Tan") {}
~OnnxTanParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};
class OnnxAtanParser : public OnnxNodeParser {
public:
OnnxAtanParser() : OnnxNodeParser("Atan") {}
~OnnxAtanParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};
class OnnxAsinParser : public OnnxNodeParser {
public:
OnnxAsinParser() : OnnxNodeParser("Asin") {}
~OnnxAsinParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};
class OnnxTanhParser : public OnnxNodeParser {
public:
OnnxTanhParser() : OnnxNodeParser("Tanh") {}
~OnnxTanhParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};
class OnnxSignParser : public OnnxNodeParser {
public:
OnnxSignParser() : OnnxNodeParser("Sign") {}
~OnnxSignParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};
class OnnxAndParser : public OnnxNodeParser {
public:
OnnxAndParser() : OnnxNodeParser("And") {}
~OnnxAndParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};
class OnnxOrParser : public OnnxNodeParser {
public:
OnnxOrParser() : OnnxNodeParser("Or") {}
~OnnxOrParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};
class OnnxNotParser : public OnnxNodeParser {
public:
OnnxNotParser() : OnnxNodeParser("Not") {}
~OnnxNotParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};
class OnnxRoundParser : public OnnxNodeParser {
public:
OnnxRoundParser() : OnnxNodeParser("Round") {}
~OnnxRoundParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};
} // namespace lite

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxBatchNormParser : public OnnxNodeParser {
public:
OnnxBatchNormParser() : OnnxNodeParser("BatchNormalization") {}
~OnnxBatchNormParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxBiasAddParser : public OnnxNodeParser {
public:
OnnxBiasAddParser() : OnnxNodeParser("BiasAdd") {}
~OnnxBiasAddParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxCastParser : public OnnxNodeParser {
public:
OnnxCastParser() : OnnxNodeParser("Cast") {}
~OnnxCastParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxClipParser : public OnnxNodeParser {
public:
OnnxClipParser() : OnnxNodeParser("Clip") {}
~OnnxClipParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxConcatParser : public OnnxNodeParser {
public:
OnnxConcatParser() : OnnxNodeParser("Concat") {}
~OnnxConcatParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxConstantOfShapeParser : public OnnxNodeParser {
public:
OnnxConstantOfShapeParser() : OnnxNodeParser("ConstantOfShape") {}
~OnnxConstantOfShapeParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxConstantParser : public OnnxNodeParser {
public:
OnnxConstantParser() : OnnxNodeParser("Constant") {}
~OnnxConstantParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -26,6 +26,7 @@ namespace lite {
class OnnxConvParser : public OnnxNodeParser {
public:
OnnxConvParser() : OnnxNodeParser("Conv") {}
~OnnxConvParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;

View File

@ -26,6 +26,7 @@ namespace lite {
class OnnxDeConvParser : public OnnxNodeParser {
public:
OnnxDeConvParser() : OnnxNodeParser("DeConv") {}
~OnnxDeConvParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxDepthToSpaceParser : public OnnxNodeParser {
public:
OnnxDepthToSpaceParser() : OnnxNodeParser("DepthToSpace") {}
~OnnxDepthToSpaceParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxDropoutParser : public OnnxNodeParser {
public:
OnnxDropoutParser() : OnnxNodeParser("Dropout") {}
~OnnxDropoutParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxEluParser : public OnnxNodeParser {
public:
OnnxEluParser() : OnnxNodeParser("Elu") {}
~OnnxEluParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxExpandParser : public OnnxNodeParser {
public:
OnnxExpandParser() : OnnxNodeParser("Expand") {}
~OnnxExpandParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxFlattenParser : public OnnxNodeParser {
public:
OnnxFlattenParser() : OnnxNodeParser("Fatten") {}
~OnnxFlattenParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxGatherParser : public OnnxNodeParser {
public:
OnnxGatherParser() : OnnxNodeParser("Gather") {}
~OnnxGatherParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxIdentityParser : public OnnxNodeParser {
public:
OnnxIdentityParser() : OnnxNodeParser("Identity") {}
~OnnxIdentityParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxInstanceNormParser : public OnnxNodeParser {
public:
OnnxInstanceNormParser() : OnnxNodeParser("InstanceNorm") {}
~OnnxInstanceNormParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxLrnParser : public OnnxNodeParser {
public:
OnnxLrnParser() : OnnxNodeParser("Lrn") {}
~OnnxLrnParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxLstmParser : public OnnxNodeParser {
public:
OnnxLstmParser() : OnnxNodeParser("LSTM") {}
~OnnxLstmParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxMatmulParser : public OnnxNodeParser {
public:
OnnxMatmulParser() : OnnxNodeParser("MatMul") {}
~OnnxMatmulParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -29,7 +29,7 @@ namespace mindspore {
namespace lite {
class OnnxNodeParser {
public:
explicit OnnxNodeParser(const std::string &nodeName) : name(nodeName) {}
explicit OnnxNodeParser(const std::string nodeName) : name(nodeName) {}
virtual ~OnnxNodeParser() = default;
@ -48,7 +48,7 @@ class OnnxNodeParser {
void Split(const std::string &src_str, std::vector<std::string> *dst_str, const std::string &chr);
const std::string &name;
const std::string name;
private:
static int opset_version_;

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxNonMaxSuppressionParser : public OnnxNodeParser {
public:
OnnxNonMaxSuppressionParser() : OnnxNodeParser("NonMaxSuppression") {}
~OnnxNonMaxSuppressionParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxOneHotParser : public OnnxNodeParser {
public:
OnnxOneHotParser() : OnnxNodeParser("OneHot") {}
~OnnxOneHotParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxPadParser : public OnnxNodeParser {
public:
OnnxPadParser() : OnnxNodeParser("Pad") {}
~OnnxPadParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxPoolParser : public OnnxNodeParser {
public:
OnnxPoolParser() : OnnxNodeParser("Pool") {}
~OnnxPoolParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxReduceParser : public OnnxNodeParser {
public:
OnnxReduceParser() : OnnxNodeParser("Reduce") {}
~OnnxReduceParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxReluParser : public OnnxNodeParser {
public:
OnnxReluParser() : OnnxNodeParser("Relu") {}
~OnnxReluParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};
@ -32,11 +33,13 @@ class OnnxReluParser : public OnnxNodeParser {
class OnnxLeakeyReluParser : public OnnxReluParser {
public:
OnnxLeakeyReluParser() : OnnxReluParser() {}
~OnnxLeakeyReluParser() override = default;
};
class OnnxPReluParser : public OnnxNodeParser {
public:
OnnxPReluParser() : OnnxNodeParser("Prelu") {}
~OnnxPReluParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxReshapeParser : public OnnxNodeParser {
public:
OnnxReshapeParser() : OnnxNodeParser("Reshape") {}
~OnnxReshapeParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxResizeParser : public OnnxNodeParser {
public:
OnnxResizeParser() : OnnxNodeParser("Resize") {}
~OnnxResizeParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxShapeParser : public OnnxNodeParser {
public:
OnnxShapeParser() : OnnxNodeParser("Shape") {}
~OnnxShapeParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxSigmoidParser : public OnnxNodeParser {
public:
OnnxSigmoidParser() : OnnxNodeParser("Sigmoid") {}
~OnnxSigmoidParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxSliceParser : public OnnxNodeParser {
public:
OnnxSliceParser() : OnnxNodeParser("Slice") {}
~OnnxSliceParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxSoftMaxParser : public OnnxNodeParser {
public:
OnnxSoftMaxParser() : OnnxNodeParser("Softmax") {}
~OnnxSoftMaxParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxSpaceToDepthParser : public OnnxNodeParser {
public:
OnnxSpaceToDepthParser() : OnnxNodeParser("SpaceToDepth") {}
~OnnxSpaceToDepthParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxSplitParser : public OnnxNodeParser {
public:
OnnxSplitParser() : OnnxNodeParser("Split") {}
~OnnxSplitParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxSqueezeParser : public OnnxNodeParser {
public:
OnnxSqueezeParser() : OnnxNodeParser("Squeeze") {}
~OnnxSqueezeParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxTileParser : public OnnxNodeParser {
public:
OnnxTileParser() : OnnxNodeParser("Tile") {}
~OnnxTileParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxTopkParser : public OnnxNodeParser {
public:
OnnxTopkParser() : OnnxNodeParser("TopK") {}
~OnnxTopkParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxTransposeParser : public OnnxNodeParser {
public:
OnnxTransposeParser() : OnnxNodeParser("Transpose") {}
~OnnxTransposeParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxUnSqueezeParser : public OnnxNodeParser {
public:
OnnxUnSqueezeParser() : OnnxNodeParser("Unsqueeze") {}
~OnnxUnSqueezeParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxUnusefulNodeParser : public OnnxNodeParser {
public:
OnnxUnusefulNodeParser() : OnnxNodeParser("UnusefulNode") {}
~OnnxUnusefulNodeParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};

View File

@ -25,6 +25,7 @@ namespace lite {
class OnnxUpsampleParser : public OnnxNodeParser {
public:
OnnxUpsampleParser() : OnnxNodeParser("Upsample") {}
~OnnxUpsampleParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
};