From b7a40262aeaf95fada3d5bce3a7adf6a375cace8 Mon Sep 17 00:00:00 2001 From: baker Date: Fri, 11 Dec 2020 10:10:23 +0800 Subject: [PATCH] add onnx model parse api --- inc/external/parser/onnx_parser.h | 6 +++--- parser/onnx/onnx_parser.cc | 4 ++-- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/inc/external/parser/onnx_parser.h b/inc/external/parser/onnx_parser.h index 20b6ebe..76bf492 100644 --- a/inc/external/parser/onnx_parser.h +++ b/inc/external/parser/onnx_parser.h @@ -14,8 +14,8 @@ * limitations under the License. */ -#ifndef INC_EXTERNAL_ACL_GRAPH_ONNX_H_ -#define INC_EXTERNAL_ACL_GRAPH_ONNX_H_ +#ifndef INC_EXTERNAL_PARSER_ONNX_PARSER_H_ +#define INC_EXTERNAL_PARSER_ONNX_PARSER_H_ #include "graph/ascend_string.h" #include "graph/ge_error_codes.h" @@ -30,4 +30,4 @@ graphStatus aclgrphParseONNXFromMem(const char *buffer, size_t size, const std::map &parser_params, ge::Graph &graph); } // namespace ge -#endif // INC_EXTERNAL_ACL_GRAPH_ONNX_H_ +#endif // INC_EXTERNAL_PARSER_ONNX_PARSER_H_ diff --git a/parser/onnx/onnx_parser.cc b/parser/onnx/onnx_parser.cc index 1b55d86..b191e24 100644 --- a/parser/onnx/onnx_parser.cc +++ b/parser/onnx/onnx_parser.cc @@ -85,7 +85,7 @@ graphStatus aclgrphParseONNX(const char *model_file, AclGrphParseUtil acl_graph_parse_util; std::shared_ptr model_parser; - if (PrepareBeforeParse(acl_graph_parse_util, parser_params, graph, model_parser) != ge::SUCCESS) { + if (PrepareBeforeParse(acl_graph_parse_util, parser_params, graph, model_parser) != ge::SUCCESS) { GELOGE(ge::FAILED, "prepare before parse failed."); return ge::FAILED; } @@ -99,7 +99,7 @@ graphStatus aclgrphParseONNX(const char *model_file, } GELOGI("Parser graph %s success.", graph.GetName().c_str()); - if (HandleAfterParse(acl_graph_parse_util, parser_params, graph) != ge::SUCCESS) { + if (HandleAfterParse(acl_graph_parse_util, parser_params, graph) != ge::SUCCESS) { GELOGE(ge::FAILED, "handle after parse failed."); return ge::FAILED; }