Browse Source

!185 fix sc

Merge pull request !185 from 陈华/development
pull/185/MERGE
i-robot Gitee 5 years ago
parent
commit
5b93b050dd
1 changed files with 3 additions and 3 deletions
  1. +3
    -3
      parser/onnx/onnx_parser.cc

+ 3
- 3
parser/onnx/onnx_parser.cc View File

@@ -68,8 +68,8 @@ graphStatus PrepareBeforeParse(AclGrphParseUtil &acl_graph_parse_util,
}

graphStatus HandleAfterParse(AclGrphParseUtil &acl_graph_parse_util,
const std::map<AscendString, AscendString> &parser_params,
ge::Graph &graph) {
const std::map<AscendString, AscendString> &parser_params,
ge::Graph &graph) {
if (acl_graph_parse_util.ParseParamsAfterGraph(graph, parser_params) != ge::SUCCESS) {
GELOGE(ge::FAILED, "Parser params after graph failed.");
return ge::FAILED;
@@ -706,4 +706,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);
}
}

Loading…
Cancel
Save