From fd654b1b45c8d76d2699d3eaba29ff6a50f79443 Mon Sep 17 00:00:00 2001 From: baker Date: Thu, 10 Dec 2020 22:09:03 +0800 Subject: [PATCH] add onnx model parse api --- parser/onnx/onnx_parser.cc | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/parser/onnx/onnx_parser.cc b/parser/onnx/onnx_parser.cc index e66b550..e2b2285 100644 --- a/parser/onnx/onnx_parser.cc +++ b/parser/onnx/onnx_parser.cc @@ -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) {