Browse Source

modify static check of onnx parser

tags/v1.1.0
liuyu 5 years ago
parent
commit
a1648b98c3
44 changed files with 75 additions and 2 deletions
  1. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_argmax_parser.h
  2. +29
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_arithmetic_operation_parser.h
  3. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_batchnorm_parser.h
  4. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_biasadd_parser.h
  5. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_cast_parser.h
  6. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_clip_parser.h
  7. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_concat_parser.h
  8. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_constant_of_shape_parser.h
  9. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_constant_parser.h
  10. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_conv_parser.h
  11. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_deconv_parser.h
  12. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_depth_to_space_parser.h
  13. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_dropout_parser.h
  14. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_elu_parser.h
  15. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_expand_parser.h
  16. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_flatten_parser.h
  17. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_gather_parser.h
  18. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_identity_parser.h
  19. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_instance_norm_parser.h
  20. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_lrn_parser.h
  21. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_lstm_parser.h
  22. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_matmul_parser.h
  23. +2
    -2
      mindspore/lite/tools/converter/parser/onnx/onnx_node_parser.h
  24. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_non_max_suppression_parser.h
  25. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_onehot_parser.h
  26. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_pad_parser.h
  27. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_pool_parser.h
  28. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_reduce_parser.h
  29. +3
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_relu_parser.h
  30. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_reshape_parser.h
  31. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_resize_parser.h
  32. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_shape_parser.h
  33. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_sigmoid_parser.h
  34. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_slice_parser.h
  35. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_softmax_parser.h
  36. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_space_to_depth_parser.h
  37. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_split_parser.h
  38. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_squeeze_parser.h
  39. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_tile_parser.h
  40. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_topk_parser.h
  41. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_transpose_parser.h
  42. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_unsqueeze_parser.h
  43. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_unuseful_node_parser.h
  44. +1
    -0
      mindspore/lite/tools/converter/parser/onnx/onnx_upsample_parser.h

+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_argmax_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxArgMaxParser : public OnnxNodeParser { class OnnxArgMaxParser : public OnnxNodeParser {
public: public:
OnnxArgMaxParser() : OnnxNodeParser("ArgMax") {} OnnxArgMaxParser() : OnnxNodeParser("ArgMax") {}
~OnnxArgMaxParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 29
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_arithmetic_operation_parser.h View File

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


class OnnxSubParser : public OnnxNodeParser { class OnnxSubParser : public OnnxNodeParser {
public: public:
OnnxSubParser() : OnnxNodeParser("Sub") {} OnnxSubParser() : OnnxNodeParser("Sub") {}
~OnnxSubParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


class OnnxMulParser : public OnnxNodeParser { class OnnxMulParser : public OnnxNodeParser {
public: public:
OnnxMulParser() : OnnxNodeParser("Mul") {} OnnxMulParser() : OnnxNodeParser("Mul") {}
~OnnxMulParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


class OnnxDivParser : public OnnxNodeParser { class OnnxDivParser : public OnnxNodeParser {
public: public:
OnnxDivParser() : OnnxNodeParser("Div") {} OnnxDivParser() : OnnxNodeParser("Div") {}
~OnnxDivParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


class OnnxMeanParser : public OnnxNodeParser { class OnnxMeanParser : public OnnxNodeParser {
public: public:
OnnxMeanParser() : OnnxNodeParser("Mean") {} OnnxMeanParser() : OnnxNodeParser("Mean") {}
~OnnxMeanParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


class OnnxPowParser : public OnnxNodeParser { class OnnxPowParser : public OnnxNodeParser {
public: public:
OnnxPowParser() : OnnxNodeParser("Power") {} OnnxPowParser() : OnnxNodeParser("Power") {}
~OnnxPowParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


class OnnxEqualParser : public OnnxNodeParser { class OnnxEqualParser : public OnnxNodeParser {
public: public:
OnnxEqualParser() : OnnxNodeParser("Equal") {} OnnxEqualParser() : OnnxNodeParser("Equal") {}
~OnnxEqualParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


class OnnxLessParser : public OnnxNodeParser { class OnnxLessParser : public OnnxNodeParser {
public: public:
OnnxLessParser() : OnnxNodeParser("Less") {} OnnxLessParser() : OnnxNodeParser("Less") {}
~OnnxLessParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


class OnnxGreaterParser : public OnnxNodeParser { class OnnxGreaterParser : public OnnxNodeParser {
public: public:
OnnxGreaterParser() : OnnxNodeParser("Greater") {} OnnxGreaterParser() : OnnxNodeParser("Greater") {}
~OnnxGreaterParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


class OnnxMinParser : public OnnxNodeParser { class OnnxMinParser : public OnnxNodeParser {
public: public:
OnnxMinParser() : OnnxNodeParser("Min") {} OnnxMinParser() : OnnxNodeParser("Min") {}
~OnnxMinParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


class OnnxEltwiseParser : public OnnxNodeParser { class OnnxEltwiseParser : public OnnxNodeParser {
public: public:
OnnxEltwiseParser() : OnnxNodeParser("Eltwise") {} OnnxEltwiseParser() : OnnxNodeParser("Eltwise") {}
~OnnxEltwiseParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


class OnnxFloorParser : public OnnxNodeParser { class OnnxFloorParser : public OnnxNodeParser {
public: public:
OnnxFloorParser() : OnnxNodeParser("Floor") {} OnnxFloorParser() : OnnxNodeParser("Floor") {}
~OnnxFloorParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


class OnnxAbsParser : public OnnxNodeParser { class OnnxAbsParser : public OnnxNodeParser {
public: public:
OnnxAbsParser() : OnnxNodeParser("Abs") {} OnnxAbsParser() : OnnxNodeParser("Abs") {}
~OnnxAbsParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


class OnnxNegParser : public OnnxNodeParser { class OnnxNegParser : public OnnxNodeParser {
public: public:
OnnxNegParser() : OnnxNodeParser("Neg") {} OnnxNegParser() : OnnxNodeParser("Neg") {}
~OnnxNegParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


class OnnxExpParser : public OnnxNodeParser { class OnnxExpParser : public OnnxNodeParser {
public: public:
OnnxExpParser() : OnnxNodeParser("Exp") {} OnnxExpParser() : OnnxNodeParser("Exp") {}
~OnnxExpParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


class OnnxCosParser : public OnnxNodeParser { class OnnxCosParser : public OnnxNodeParser {
public: public:
OnnxCosParser() : OnnxNodeParser("Cos") {} OnnxCosParser() : OnnxNodeParser("Cos") {}
~OnnxCosParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


class OnnxSinParser : public OnnxNodeParser { class OnnxSinParser : public OnnxNodeParser {
public: public:
OnnxSinParser() : OnnxNodeParser("Sin") {} OnnxSinParser() : OnnxNodeParser("Sin") {}
~OnnxSinParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


class OnnxSqrtParser : public OnnxNodeParser { class OnnxSqrtParser : public OnnxNodeParser {
public: public:
OnnxSqrtParser() : OnnxNodeParser("Sqrt") {} OnnxSqrtParser() : OnnxNodeParser("Sqrt") {}
~OnnxSqrtParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


class OnnxCeilParser : public OnnxNodeParser { class OnnxCeilParser : public OnnxNodeParser {
public: public:
OnnxCeilParser() : OnnxNodeParser("Ceil") {} OnnxCeilParser() : OnnxNodeParser("Ceil") {}
~OnnxCeilParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


class OnnxLogParser : public OnnxNodeParser { class OnnxLogParser : public OnnxNodeParser {
public: public:
OnnxLogParser() : OnnxNodeParser("Log") {} OnnxLogParser() : OnnxNodeParser("Log") {}
~OnnxLogParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


class OnnxTanParser : public OnnxNodeParser { class OnnxTanParser : public OnnxNodeParser {
public: public:
OnnxTanParser() : OnnxNodeParser("Tan") {} OnnxTanParser() : OnnxNodeParser("Tan") {}
~OnnxTanParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


class OnnxAtanParser : public OnnxNodeParser { class OnnxAtanParser : public OnnxNodeParser {
public: public:
OnnxAtanParser() : OnnxNodeParser("Atan") {} OnnxAtanParser() : OnnxNodeParser("Atan") {}
~OnnxAtanParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


class OnnxAsinParser : public OnnxNodeParser { class OnnxAsinParser : public OnnxNodeParser {
public: public:
OnnxAsinParser() : OnnxNodeParser("Asin") {} OnnxAsinParser() : OnnxNodeParser("Asin") {}
~OnnxAsinParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


class OnnxTanhParser : public OnnxNodeParser { class OnnxTanhParser : public OnnxNodeParser {
public: public:
OnnxTanhParser() : OnnxNodeParser("Tanh") {} OnnxTanhParser() : OnnxNodeParser("Tanh") {}
~OnnxTanhParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


class OnnxSignParser : public OnnxNodeParser { class OnnxSignParser : public OnnxNodeParser {
public: public:
OnnxSignParser() : OnnxNodeParser("Sign") {} OnnxSignParser() : OnnxNodeParser("Sign") {}
~OnnxSignParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


class OnnxAndParser : public OnnxNodeParser { class OnnxAndParser : public OnnxNodeParser {
public: public:
OnnxAndParser() : OnnxNodeParser("And") {} OnnxAndParser() : OnnxNodeParser("And") {}
~OnnxAndParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


class OnnxOrParser : public OnnxNodeParser { class OnnxOrParser : public OnnxNodeParser {
public: public:
OnnxOrParser() : OnnxNodeParser("Or") {} OnnxOrParser() : OnnxNodeParser("Or") {}
~OnnxOrParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


class OnnxNotParser : public OnnxNodeParser { class OnnxNotParser : public OnnxNodeParser {
public: public:
OnnxNotParser() : OnnxNodeParser("Not") {} OnnxNotParser() : OnnxNodeParser("Not") {}
~OnnxNotParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


class OnnxRoundParser : public OnnxNodeParser { class OnnxRoundParser : public OnnxNodeParser {
public: public:
OnnxRoundParser() : OnnxNodeParser("Round") {} OnnxRoundParser() : OnnxNodeParser("Round") {}
~OnnxRoundParser() override = default;
STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };
} // namespace lite } // namespace lite


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_batchnorm_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxBatchNormParser : public OnnxNodeParser { class OnnxBatchNormParser : public OnnxNodeParser {
public: public:
OnnxBatchNormParser() : OnnxNodeParser("BatchNormalization") {} OnnxBatchNormParser() : OnnxNodeParser("BatchNormalization") {}
~OnnxBatchNormParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_biasadd_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxBiasAddParser : public OnnxNodeParser { class OnnxBiasAddParser : public OnnxNodeParser {
public: public:
OnnxBiasAddParser() : OnnxNodeParser("BiasAdd") {} OnnxBiasAddParser() : OnnxNodeParser("BiasAdd") {}
~OnnxBiasAddParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_cast_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxCastParser : public OnnxNodeParser { class OnnxCastParser : public OnnxNodeParser {
public: public:
OnnxCastParser() : OnnxNodeParser("Cast") {} OnnxCastParser() : OnnxNodeParser("Cast") {}
~OnnxCastParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_clip_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxClipParser : public OnnxNodeParser { class OnnxClipParser : public OnnxNodeParser {
public: public:
OnnxClipParser() : OnnxNodeParser("Clip") {} OnnxClipParser() : OnnxNodeParser("Clip") {}
~OnnxClipParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_concat_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxConcatParser : public OnnxNodeParser { class OnnxConcatParser : public OnnxNodeParser {
public: public:
OnnxConcatParser() : OnnxNodeParser("Concat") {} OnnxConcatParser() : OnnxNodeParser("Concat") {}
~OnnxConcatParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_constant_of_shape_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxConstantOfShapeParser : public OnnxNodeParser { class OnnxConstantOfShapeParser : public OnnxNodeParser {
public: public:
OnnxConstantOfShapeParser() : OnnxNodeParser("ConstantOfShape") {} OnnxConstantOfShapeParser() : OnnxNodeParser("ConstantOfShape") {}
~OnnxConstantOfShapeParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_constant_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxConstantParser : public OnnxNodeParser { class OnnxConstantParser : public OnnxNodeParser {
public: public:
OnnxConstantParser() : OnnxNodeParser("Constant") {} OnnxConstantParser() : OnnxNodeParser("Constant") {}
~OnnxConstantParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_conv_parser.h View File

@@ -26,6 +26,7 @@ namespace lite {
class OnnxConvParser : public OnnxNodeParser { class OnnxConvParser : public OnnxNodeParser {
public: public:
OnnxConvParser() : OnnxNodeParser("Conv") {} OnnxConvParser() : OnnxNodeParser("Conv") {}
~OnnxConvParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;




+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_deconv_parser.h View File

@@ -26,6 +26,7 @@ namespace lite {
class OnnxDeConvParser : public OnnxNodeParser { class OnnxDeConvParser : public OnnxNodeParser {
public: public:
OnnxDeConvParser() : OnnxNodeParser("DeConv") {} OnnxDeConvParser() : OnnxNodeParser("DeConv") {}
~OnnxDeConvParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;




+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_depth_to_space_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxDepthToSpaceParser : public OnnxNodeParser { class OnnxDepthToSpaceParser : public OnnxNodeParser {
public: public:
OnnxDepthToSpaceParser() : OnnxNodeParser("DepthToSpace") {} OnnxDepthToSpaceParser() : OnnxNodeParser("DepthToSpace") {}
~OnnxDepthToSpaceParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_dropout_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxDropoutParser : public OnnxNodeParser { class OnnxDropoutParser : public OnnxNodeParser {
public: public:
OnnxDropoutParser() : OnnxNodeParser("Dropout") {} OnnxDropoutParser() : OnnxNodeParser("Dropout") {}
~OnnxDropoutParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_elu_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxEluParser : public OnnxNodeParser { class OnnxEluParser : public OnnxNodeParser {
public: public:
OnnxEluParser() : OnnxNodeParser("Elu") {} OnnxEluParser() : OnnxNodeParser("Elu") {}
~OnnxEluParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_expand_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxExpandParser : public OnnxNodeParser { class OnnxExpandParser : public OnnxNodeParser {
public: public:
OnnxExpandParser() : OnnxNodeParser("Expand") {} OnnxExpandParser() : OnnxNodeParser("Expand") {}
~OnnxExpandParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_flatten_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxFlattenParser : public OnnxNodeParser { class OnnxFlattenParser : public OnnxNodeParser {
public: public:
OnnxFlattenParser() : OnnxNodeParser("Fatten") {} OnnxFlattenParser() : OnnxNodeParser("Fatten") {}
~OnnxFlattenParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_gather_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxGatherParser : public OnnxNodeParser { class OnnxGatherParser : public OnnxNodeParser {
public: public:
OnnxGatherParser() : OnnxNodeParser("Gather") {} OnnxGatherParser() : OnnxNodeParser("Gather") {}
~OnnxGatherParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_identity_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxIdentityParser : public OnnxNodeParser { class OnnxIdentityParser : public OnnxNodeParser {
public: public:
OnnxIdentityParser() : OnnxNodeParser("Identity") {} OnnxIdentityParser() : OnnxNodeParser("Identity") {}
~OnnxIdentityParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_instance_norm_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxInstanceNormParser : public OnnxNodeParser { class OnnxInstanceNormParser : public OnnxNodeParser {
public: public:
OnnxInstanceNormParser() : OnnxNodeParser("InstanceNorm") {} OnnxInstanceNormParser() : OnnxNodeParser("InstanceNorm") {}
~OnnxInstanceNormParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_lrn_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxLrnParser : public OnnxNodeParser { class OnnxLrnParser : public OnnxNodeParser {
public: public:
OnnxLrnParser() : OnnxNodeParser("Lrn") {} OnnxLrnParser() : OnnxNodeParser("Lrn") {}
~OnnxLrnParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_lstm_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxLstmParser : public OnnxNodeParser { class OnnxLstmParser : public OnnxNodeParser {
public: public:
OnnxLstmParser() : OnnxNodeParser("LSTM") {} OnnxLstmParser() : OnnxNodeParser("LSTM") {}
~OnnxLstmParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_matmul_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxMatmulParser : public OnnxNodeParser { class OnnxMatmulParser : public OnnxNodeParser {
public: public:
OnnxMatmulParser() : OnnxNodeParser("MatMul") {} OnnxMatmulParser() : OnnxNodeParser("MatMul") {}
~OnnxMatmulParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 2
- 2
mindspore/lite/tools/converter/parser/onnx/onnx_node_parser.h View File

@@ -29,7 +29,7 @@ namespace mindspore {
namespace lite { namespace lite {
class OnnxNodeParser { class OnnxNodeParser {
public: public:
explicit OnnxNodeParser(const std::string &nodeName) : name(nodeName) {}
explicit OnnxNodeParser(const std::string nodeName) : name(nodeName) {}


virtual ~OnnxNodeParser() = default; 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); 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: private:
static int opset_version_; static int opset_version_;


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_non_max_suppression_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxNonMaxSuppressionParser : public OnnxNodeParser { class OnnxNonMaxSuppressionParser : public OnnxNodeParser {
public: public:
OnnxNonMaxSuppressionParser() : OnnxNodeParser("NonMaxSuppression") {} OnnxNonMaxSuppressionParser() : OnnxNodeParser("NonMaxSuppression") {}
~OnnxNonMaxSuppressionParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_onehot_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxOneHotParser : public OnnxNodeParser { class OnnxOneHotParser : public OnnxNodeParser {
public: public:
OnnxOneHotParser() : OnnxNodeParser("OneHot") {} OnnxOneHotParser() : OnnxNodeParser("OneHot") {}
~OnnxOneHotParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_pad_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxPadParser : public OnnxNodeParser { class OnnxPadParser : public OnnxNodeParser {
public: public:
OnnxPadParser() : OnnxNodeParser("Pad") {} OnnxPadParser() : OnnxNodeParser("Pad") {}
~OnnxPadParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_pool_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxPoolParser : public OnnxNodeParser { class OnnxPoolParser : public OnnxNodeParser {
public: public:
OnnxPoolParser() : OnnxNodeParser("Pool") {} OnnxPoolParser() : OnnxNodeParser("Pool") {}
~OnnxPoolParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_reduce_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxReduceParser : public OnnxNodeParser { class OnnxReduceParser : public OnnxNodeParser {
public: public:
OnnxReduceParser() : OnnxNodeParser("Reduce") {} OnnxReduceParser() : OnnxNodeParser("Reduce") {}
~OnnxReduceParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 3
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_relu_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxReluParser : public OnnxNodeParser { class OnnxReluParser : public OnnxNodeParser {
public: public:
OnnxReluParser() : OnnxNodeParser("Relu") {} OnnxReluParser() : OnnxNodeParser("Relu") {}
~OnnxReluParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; 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 { class OnnxLeakeyReluParser : public OnnxReluParser {
public: public:
OnnxLeakeyReluParser() : OnnxReluParser() {} OnnxLeakeyReluParser() : OnnxReluParser() {}
~OnnxLeakeyReluParser() override = default;
}; };


class OnnxPReluParser : public OnnxNodeParser { class OnnxPReluParser : public OnnxNodeParser {
public: public:
OnnxPReluParser() : OnnxNodeParser("Prelu") {} OnnxPReluParser() : OnnxNodeParser("Prelu") {}
~OnnxPReluParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_reshape_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxReshapeParser : public OnnxNodeParser { class OnnxReshapeParser : public OnnxNodeParser {
public: public:
OnnxReshapeParser() : OnnxNodeParser("Reshape") {} OnnxReshapeParser() : OnnxNodeParser("Reshape") {}
~OnnxReshapeParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_resize_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxResizeParser : public OnnxNodeParser { class OnnxResizeParser : public OnnxNodeParser {
public: public:
OnnxResizeParser() : OnnxNodeParser("Resize") {} OnnxResizeParser() : OnnxNodeParser("Resize") {}
~OnnxResizeParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_shape_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxShapeParser : public OnnxNodeParser { class OnnxShapeParser : public OnnxNodeParser {
public: public:
OnnxShapeParser() : OnnxNodeParser("Shape") {} OnnxShapeParser() : OnnxNodeParser("Shape") {}
~OnnxShapeParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_sigmoid_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxSigmoidParser : public OnnxNodeParser { class OnnxSigmoidParser : public OnnxNodeParser {
public: public:
OnnxSigmoidParser() : OnnxNodeParser("Sigmoid") {} OnnxSigmoidParser() : OnnxNodeParser("Sigmoid") {}
~OnnxSigmoidParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_slice_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxSliceParser : public OnnxNodeParser { class OnnxSliceParser : public OnnxNodeParser {
public: public:
OnnxSliceParser() : OnnxNodeParser("Slice") {} OnnxSliceParser() : OnnxNodeParser("Slice") {}
~OnnxSliceParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_softmax_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxSoftMaxParser : public OnnxNodeParser { class OnnxSoftMaxParser : public OnnxNodeParser {
public: public:
OnnxSoftMaxParser() : OnnxNodeParser("Softmax") {} OnnxSoftMaxParser() : OnnxNodeParser("Softmax") {}
~OnnxSoftMaxParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_space_to_depth_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxSpaceToDepthParser : public OnnxNodeParser { class OnnxSpaceToDepthParser : public OnnxNodeParser {
public: public:
OnnxSpaceToDepthParser() : OnnxNodeParser("SpaceToDepth") {} OnnxSpaceToDepthParser() : OnnxNodeParser("SpaceToDepth") {}
~OnnxSpaceToDepthParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_split_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxSplitParser : public OnnxNodeParser { class OnnxSplitParser : public OnnxNodeParser {
public: public:
OnnxSplitParser() : OnnxNodeParser("Split") {} OnnxSplitParser() : OnnxNodeParser("Split") {}
~OnnxSplitParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_squeeze_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxSqueezeParser : public OnnxNodeParser { class OnnxSqueezeParser : public OnnxNodeParser {
public: public:
OnnxSqueezeParser() : OnnxNodeParser("Squeeze") {} OnnxSqueezeParser() : OnnxNodeParser("Squeeze") {}
~OnnxSqueezeParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_tile_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxTileParser : public OnnxNodeParser { class OnnxTileParser : public OnnxNodeParser {
public: public:
OnnxTileParser() : OnnxNodeParser("Tile") {} OnnxTileParser() : OnnxNodeParser("Tile") {}
~OnnxTileParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_topk_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxTopkParser : public OnnxNodeParser { class OnnxTopkParser : public OnnxNodeParser {
public: public:
OnnxTopkParser() : OnnxNodeParser("TopK") {} OnnxTopkParser() : OnnxNodeParser("TopK") {}
~OnnxTopkParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_transpose_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxTransposeParser : public OnnxNodeParser { class OnnxTransposeParser : public OnnxNodeParser {
public: public:
OnnxTransposeParser() : OnnxNodeParser("Transpose") {} OnnxTransposeParser() : OnnxNodeParser("Transpose") {}
~OnnxTransposeParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_unsqueeze_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxUnSqueezeParser : public OnnxNodeParser { class OnnxUnSqueezeParser : public OnnxNodeParser {
public: public:
OnnxUnSqueezeParser() : OnnxNodeParser("Unsqueeze") {} OnnxUnSqueezeParser() : OnnxNodeParser("Unsqueeze") {}
~OnnxUnSqueezeParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_unuseful_node_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxUnusefulNodeParser : public OnnxNodeParser { class OnnxUnusefulNodeParser : public OnnxNodeParser {
public: public:
OnnxUnusefulNodeParser() : OnnxNodeParser("UnusefulNode") {} OnnxUnusefulNodeParser() : OnnxNodeParser("UnusefulNode") {}
~OnnxUnusefulNodeParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


+ 1
- 0
mindspore/lite/tools/converter/parser/onnx/onnx_upsample_parser.h View File

@@ -25,6 +25,7 @@ namespace lite {
class OnnxUpsampleParser : public OnnxNodeParser { class OnnxUpsampleParser : public OnnxNodeParser {
public: public:
OnnxUpsampleParser() : OnnxNodeParser("Upsample") {} OnnxUpsampleParser() : OnnxNodeParser("Upsample") {}
~OnnxUpsampleParser() override = default;


STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override; STATUS Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node, schema::CNodeT *op) override;
}; };


Loading…
Cancel
Save