Browse Source

add onnx model parse api

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

@@ -41,7 +41,7 @@
namespace ge {
graphStatus PrepareBeforeParse(AclGrphParseUtil &acl_graph_parse_util,
const std::map<AscendString, AscendString> &parser_params,
ge::Graph &graph, domi::ModelParser *model_parser) {
ge::Graph &graph, std::shared_ptr<domi::ModelParser> model_parser) {
GetParserContext().type = domi::ONNX;
std::map<string, string> options;
options.insert(std::pair<string, string>(string(ge::FRAMEWORK_TYPE), to_string(ge::ONNX)));
@@ -83,7 +83,7 @@ graphStatus aclgrphParseONNX(const char *model_file,
GE_CHECK_NOTNULL(model_file);
// load custom plugin so and proto
AclGrphParseUtil acl_graph_parse_util;
domi::ModelParser *model_parser = nullptr;
std::shared_ptr<domi::ModelParser> model_parser;

if (PrepareBeforeParse(acl_graph_parse_util, parser_params, graph, model_parser) != ge::SUCCESS) {
GELOGE(ge::FAILED, "prepare before parse failed.");
@@ -113,7 +113,7 @@ graphStatus aclgrphParseONNXFromMem(const char *buffer, size_t size,
GE_CHECK_NOTNULL(buffer);
// load custom plugin so and proto
AclGrphParseUtil acl_graph_parse_util;
domi::ModelParser *model_parser = nullptr;
std::shared_ptr<domi::ModelParser> model_parser;

if (PrepareBeforeParse(acl_graph_parse_util, parser_params, graph, model_parser) != ge::SUCCESS) {
GELOGE(ge::FAILED, "prepare before parse failed.");


Loading…
Cancel
Save