Browse Source

add onnx model parse api

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

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

@@ -630,11 +630,12 @@ Status OnnxModelParser::Parse(const char *file, ge::Graph &graph) {
GELOGE(FAILED, "get model from file failed.");
return FAILED;
}
ret = RealParse(onnx_model, graph)
ret = RealParse(onnx_model, graph);
if (ret != SUCCESS) {
GELOGE(FAILED, "parse model failed.");
return FAILED;
}
return SUCCESS;
}

Status OnnxModelParser::ParseFromMemory(const char *data, uint32_t size, ge::Graph &graph) {
@@ -649,6 +650,7 @@ Status OnnxModelParser::ParseFromMemory(const char *data, uint32_t size, ge::Gra
GELOGE(FAILED, "parse model failed.");
return FAILED;
}
return SUCCESS;
}

Status OnnxModelParser::ToJson(const char *model_file, const char *json_file) {


Loading…
Cancel
Save