|
|
|
@@ -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."); |
|
|
|
|