From 0b196449efcc594c7799667f073c99a685e85626 Mon Sep 17 00:00:00 2001 From: baker Date: Tue, 5 Jan 2021 14:28:38 +0800 Subject: [PATCH] fix bug --- inc/external/parser/onnx_parser.h | 6 ++++-- parser/onnx/onnx_parser.cc | 10 +++++++--- 2 files changed, 11 insertions(+), 5 deletions(-) diff --git a/inc/external/parser/onnx_parser.h b/inc/external/parser/onnx_parser.h index 7678218..116fc72 100644 --- a/inc/external/parser/onnx_parser.h +++ b/inc/external/parser/onnx_parser.h @@ -27,11 +27,13 @@ namespace ge { graphStatus aclgrphParseONNX(const char *model_file, - std::map &parser_params, + const std::map &parser_params, ge::Graph &graph); graphStatus aclgrphParseONNXFromMem(const char *buffer, size_t buffer_size, - std::map &parser_params, + const std::map &parser_params, ge::Graph &graph); } // namespace ge diff --git a/parser/onnx/onnx_parser.cc b/parser/onnx/onnx_parser.cc index 4925f69..5e51446 100644 --- a/parser/onnx/onnx_parser.cc +++ b/parser/onnx/onnx_parser.cc @@ -40,7 +40,9 @@ namespace ge { graphStatus aclgrphParseONNX(const char *model_file, - std::map &parser_params, ge::Graph &graph) { + const std::map &parser_params, + ge::Graph &graph) { GE_CHECK_NOTNULL(model_file); GetParserContext().type = domi::ONNX; std::map options; @@ -86,7 +88,9 @@ graphStatus aclgrphParseONNX(const char *model_file, } graphStatus aclgrphParseONNXFromMem(const char *buffer, size_t buffer_size, - std::map &parser_params, ge::Graph &graph) { + const std::map &parser_params, + ge::Graph &graph) { GE_CHECK_NOTNULL(buffer); GetParserContext().type = domi::ONNX; std::map options; @@ -672,4 +676,4 @@ ge::DataType OnnxModelParser::ConvertToGeDataType(const uint32_t type) { namespace domi { REGISTER_MODEL_PARSER_CREATOR(ONNX, ge::OnnxModelParser); REGISTER_WEIGHTS_PARSER_CREATOR(ONNX, ge::OnnxWeightsParser); -} \ No newline at end of file +}