forked from mindspore-Ecosystem/mindspore
!7995 modify static check of onnx parser
Merge pull request !7995 from lyvette/master2
This commit is contained in:
commit
00469ba584
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
|
||||
|
|
|
@ -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;
|
||||
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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_;
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
|
@ -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;
|
||||
};
|
||||
|
|
Loading…
Reference in New Issue