|
|
@ -86,13 +86,10 @@ Graph Serialization::LoadModel(const void *model_data, size_t data_size, ModelTy
|
|
|
|
|
|
|
|
|
|
|
|
Graph Serialization::LoadModel(const std::string &file, ModelType model_type) {
|
|
|
|
Graph Serialization::LoadModel(const std::string &file, ModelType model_type) {
|
|
|
|
if (model_type == kMindIR) {
|
|
|
|
if (model_type == kMindIR) {
|
|
|
|
FuncGraphPtr anf_graph = nullptr;
|
|
|
|
FuncGraphPtr anf_graph = LoadMindIR(file);
|
|
|
|
try {
|
|
|
|
if (anf_graph == nullptr) {
|
|
|
|
anf_graph = LoadMindIR(file);
|
|
|
|
MS_LOG(EXCEPTION) << "Load model failed.";
|
|
|
|
} catch (const std::exception &) {
|
|
|
|
|
|
|
|
MS_LOG(EXCEPTION) << "Load MindIR failed.";
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
return Graph(std::make_shared<Graph::GraphData>(anf_graph, kMindIR));
|
|
|
|
return Graph(std::make_shared<Graph::GraphData>(anf_graph, kMindIR));
|
|
|
|
} else if (model_type == kOM) {
|
|
|
|
} else if (model_type == kOM) {
|
|
|
|
Buffer data = ReadFile(file);
|
|
|
|
Buffer data = ReadFile(file);
|
|
|
|