wangwenhua1@huawei.com
5 years ago
1 changed files with 2 additions and 2 deletions
Unified View
Diff Options
-
+2 -2ge/session/omg.cc
+ 2
- 2
ge/session/omg.cc
View File
| @@ -874,7 +874,7 @@ FMK_FUNC_HOST_VISIBILITY void PrintModelInfo(ge::proto::ModelDef *model_def) { | |||||
| auto model_attr_map = model_def->mutable_attr(); | auto model_attr_map = model_def->mutable_attr(); | ||||
| // system info | // system info | ||||
| iter = model_attr_map->find(ATTR_MODEL_ATC_VERSION); | |||||
| auto iter = model_attr_map->find(ATTR_MODEL_ATC_VERSION); | |||||
| auto atc_version = (iter != model_attr_map->end()) ? iter->second.s() : ""; | auto atc_version = (iter != model_attr_map->end()) ? iter->second.s() : ""; | ||||
| iter = model_attr_map->find("soc_version"); | iter = model_attr_map->find("soc_version"); | ||||
| auto soc_version = (iter != model_attr_map->end()) ? iter->second.s() : ""; | auto soc_version = (iter != model_attr_map->end()) ? iter->second.s() : ""; | ||||
| @@ -889,7 +889,7 @@ FMK_FUNC_HOST_VISIBILITY void PrintModelInfo(ge::proto::ModelDef *model_def) { | |||||
| << "[" << framework_type << "]." << std::endl; | << "[" << framework_type << "]." << std::endl; | ||||
| // resource info | // resource info | ||||
| auto iter = model_attr_map->find(ATTR_MODEL_MEMORY_SIZE); | |||||
| iter = model_attr_map->find(ATTR_MODEL_MEMORY_SIZE); | |||||
| auto memory_size = (iter != model_attr_map->end()) ? iter->second.i() : -1; | auto memory_size = (iter != model_attr_map->end()) ? iter->second.i() : -1; | ||||
| iter = model_attr_map->find(ATTR_MODEL_WEIGHT_SIZE); | iter = model_attr_map->find(ATTR_MODEL_WEIGHT_SIZE); | ||||
| auto weight_size = (iter != model_attr_map->end()) ? iter->second.i() : -1; | auto weight_size = (iter != model_attr_map->end()) ? iter->second.i() : -1; | ||||