|
@@ -40,7 +40,9 @@ |
|
|
|
|
|
|
|
|
namespace ge { |
|
|
namespace ge { |
|
|
graphStatus aclgrphParseONNX(const char *model_file, |
|
|
graphStatus aclgrphParseONNX(const char *model_file, |
|
|
std::map<AscendString, AscendString> &parser_params, ge::Graph &graph) { |
|
|
|
|
|
|
|
|
const std::map<AscendString, |
|
|
|
|
|
AscendString> &parser_params, |
|
|
|
|
|
ge::Graph &graph) { |
|
|
GE_CHECK_NOTNULL(model_file); |
|
|
GE_CHECK_NOTNULL(model_file); |
|
|
GetParserContext().type = domi::ONNX; |
|
|
GetParserContext().type = domi::ONNX; |
|
|
std::map<string, string> options; |
|
|
std::map<string, string> options; |
|
@@ -86,7 +88,9 @@ graphStatus aclgrphParseONNX(const char *model_file, |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
graphStatus aclgrphParseONNXFromMem(const char *buffer, size_t buffer_size, |
|
|
graphStatus aclgrphParseONNXFromMem(const char *buffer, size_t buffer_size, |
|
|
std::map<AscendString, AscendString> &parser_params, ge::Graph &graph) { |
|
|
|
|
|
|
|
|
const std::map<AscendString, |
|
|
|
|
|
AscendString> &parser_params, |
|
|
|
|
|
ge::Graph &graph) { |
|
|
GE_CHECK_NOTNULL(buffer); |
|
|
GE_CHECK_NOTNULL(buffer); |
|
|
GetParserContext().type = domi::ONNX; |
|
|
GetParserContext().type = domi::ONNX; |
|
|
std::map<string, string> options; |
|
|
std::map<string, string> options; |
|
@@ -672,4 +676,4 @@ ge::DataType OnnxModelParser::ConvertToGeDataType(const uint32_t type) { |
|
|
namespace domi { |
|
|
namespace domi { |
|
|
REGISTER_MODEL_PARSER_CREATOR(ONNX, ge::OnnxModelParser); |
|
|
REGISTER_MODEL_PARSER_CREATOR(ONNX, ge::OnnxModelParser); |
|
|
REGISTER_WEIGHTS_PARSER_CREATOR(ONNX, ge::OnnxWeightsParser); |
|
|
REGISTER_WEIGHTS_PARSER_CREATOR(ONNX, ge::OnnxWeightsParser); |
|
|
} |
|
|
|
|
|
|
|
|
} |