Browse Source

Pre Merge pull request !174 from 陈华/development

pull/174/MERGE
陈华 Gitee 5 years ago
parent
commit
9f0ddc9fe6
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