|
|
|
@@ -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) { |
|
|
|
|