|
|
|
@@ -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; |
|
|
|
|