From 8a4dbc9424095ea6576a9b43587dec4843b1aff9 Mon Sep 17 00:00:00 2001
From: baker <chenhua26@hisilicon.com>
Date: Wed, 9 Dec 2020 21:59:24 +0800
Subject: [PATCH] add onnx parse api

---
 inc/framework/common/ge_types.h             |  1 +
 inc/framework/omg/parser/model_parser.h     | 11 +++++++++++
 inc/framework/omg/parser/parser_inner_ctx.h |  2 +-
 3 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/inc/framework/common/ge_types.h b/inc/framework/common/ge_types.h
index fb1f0be1..254dbada 100644
--- a/inc/framework/common/ge_types.h
+++ b/inc/framework/common/ge_types.h
@@ -37,6 +37,7 @@ enum FrameworkType {
   MINDSPORE = 1,
   TENSORFLOW = 3,
   ANDROID_NN,
+  ONNX,
   FRAMEWORK_RESERVED,
 };
 
diff --git a/inc/framework/omg/parser/model_parser.h b/inc/framework/omg/parser/model_parser.h
index 20bfcef4..bb3a04b9 100644
--- a/inc/framework/omg/parser/model_parser.h
+++ b/inc/framework/omg/parser/model_parser.h
@@ -65,6 +65,17 @@ class ModelParser {
   virtual Status ParseFromMemory(const char *data, uint32_t size, ge::ComputeGraphPtr &graph) = 0;
 
   /**
+   * @ingroup domi_omg
+   * @brief Parse relevant data from memory and save it to graph
+   * @param [in] input Model file memory data
+   * @param [in|out] graph A graph for saving the model information after analysis
+   * @return SUCCESS
+   * @return FAILED
+   * @author
+   */
+  virtual Status ParseFromMemory(const char *data, uint32_t size, ge::Graph &graph) = 0;
+
+  /**
   * @ingroup domi_omg
   * @brief Analyze network model data
   * @param [in] proto  network model
diff --git a/inc/framework/omg/parser/parser_inner_ctx.h b/inc/framework/omg/parser/parser_inner_ctx.h
index f24e2639..5d91bd46 100644
--- a/inc/framework/omg/parser/parser_inner_ctx.h
+++ b/inc/framework/omg/parser/parser_inner_ctx.h
@@ -59,7 +59,7 @@ struct ParserContext {
   bool train_flag = false;
   domi::domiTensorFormat_t format = domi::DOMI_TENSOR_ND;
   domi::FrameworkType type = domi::FRAMEWORK_RESERVED;
-  RunMode run_mode = ONLY_PRE_CHECK;
+  RunMode run_mode = GEN_OM_MODEL;
   // save caffe custom proto path, used by caffe parse
   std::string custom_proto_path;
   // save caffe proto path, used by caffe parse