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; }