|
|
|
@@ -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 |
|
|
|
|