Browse Source

add onnx model parse api

pull/134/head
baker 5 years ago
parent
commit
5c5c138941
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

@@ -111,7 +111,7 @@ graphStatus aclgrphParseONNXFromMem(const char *buffer, size_t size,
GE_CHECK_NOTNULL(model_parser);

// parse caffe model_file to GE graph
ge::graphStatus ret = model_parser->Parse(buffer, (uint32_t)size, graph);
ge::graphStatus ret = model_parser->ParseFromMemory(buffer, (uint32_t)size, graph);
if (ret != ge::SUCCESS) {
GELOGE(ret, "Parser graph %s failed.", graph.GetName().c_str());
return ge::FAILED;
@@ -506,7 +506,7 @@ Status OnnxModelParser::GetModelFromFile(const char *file, ge::onnx::ModelProto
GELOGI("File path is %s.", file);

// 1. Get graph from onnx model file.
if (!ge::parser::ReadProtoFromBinaryFile(file, onnx_model)) {
if (!ge::parser::ReadProtoFromBinaryFile(file, &onnx_model)) {
ErrorManager::GetInstance().ATCReportErrMessage(
"E19021", {"reason"}, {"Read onnx model file failed."});
GELOGE(PARAM_INVALID, "Read onnx model file failed.");
@@ -645,7 +645,7 @@ Status OnnxModelParser::ParseFromMemory(const char *data, uint32_t size, ge::Gra
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;


Loading…
Cancel
Save