Browse Source

add onnx model parse api

pull/134/head
baker 5 years ago
parent
commit
169d688d9c
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      parser/onnx/onnx_parser.cc

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

@@ -41,7 +41,7 @@
namespace ge {
graphStatus PrepareBeforeParse(AclGrphParseUtil &acl_graph_parse_util,
const std::map<AscendString, AscendString> &parser_params,
ge::Graph &graph, std::shared_ptr<domi::ModelParser> model_parser) {
ge::Graph &graph, std::shared_ptr<domi::ModelParser> &model_parser) {
GetParserContext().type = domi::ONNX;
std::map<string, string> options;
options.insert(std::pair<string, string>(string(ge::FRAMEWORK_TYPE), to_string(ge::ONNX)));


Loading…
Cancel
Save