From d6e82cb6a185a3396ae935f0a8308e23663d3c37 Mon Sep 17 00:00:00 2001 From: baker Date: Fri, 11 Dec 2020 09:24:10 +0800 Subject: [PATCH] add onnx model parse api --- parser/onnx/onnx_parser.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/parser/onnx/onnx_parser.cc b/parser/onnx/onnx_parser.cc index fd49ac1..6783fda 100644 --- a/parser/onnx/onnx_parser.cc +++ b/parser/onnx/onnx_parser.cc @@ -41,7 +41,7 @@ namespace ge { graphStatus PrepareBeforeParse(AclGrphParseUtil &acl_graph_parse_util, const std::map &parser_params, - ge::Graph &graph, domi::ModelParser *model_parser) { + ge::Graph &graph, std::shared_ptr model_parser) { GetParserContext().type = domi::ONNX; std::map options; options.insert(std::pair(string(ge::FRAMEWORK_TYPE), to_string(ge::ONNX))); @@ -83,7 +83,7 @@ graphStatus aclgrphParseONNX(const char *model_file, GE_CHECK_NOTNULL(model_file); // load custom plugin so and proto AclGrphParseUtil acl_graph_parse_util; - domi::ModelParser *model_parser = nullptr; + std::shared_ptr model_parser; if (PrepareBeforeParse(acl_graph_parse_util, parser_params, graph, model_parser) != ge::SUCCESS) { GELOGE(ge::FAILED, "prepare before parse failed."); @@ -113,7 +113,7 @@ graphStatus aclgrphParseONNXFromMem(const char *buffer, size_t size, GE_CHECK_NOTNULL(buffer); // load custom plugin so and proto AclGrphParseUtil acl_graph_parse_util; - domi::ModelParser *model_parser = nullptr; + std::shared_ptr model_parser; if (PrepareBeforeParse(acl_graph_parse_util, parser_params, graph, model_parser) != ge::SUCCESS) { GELOGE(ge::FAILED, "prepare before parse failed.");