From 93600779f7beedbeeaaf9f53c42c8f58e34ccab8 Mon Sep 17 00:00:00 2001 From: baker Date: Thu, 10 Dec 2020 11:38:11 +0800 Subject: [PATCH] add onnx model parse api --- inc/framework/omg/parser/model_parser.h | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/inc/framework/omg/parser/model_parser.h b/inc/framework/omg/parser/model_parser.h index 20bfcef4..201b12d8 100644 --- a/inc/framework/omg/parser/model_parser.h +++ b/inc/framework/omg/parser/model_parser.h @@ -66,6 +66,17 @@ class ModelParser { /** * @ingroup domi_omg + * @brief Parse relevant data from memory and save it to graph + * @param [in] input Model file memory data and size + * @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 * @param [in|out] graph Save the network information after analysis