| @@ -872,6 +872,7 @@ void GetGroupName(ge::proto::ModelDef &model_def) { | |||||
| FMK_FUNC_HOST_VISIBILITY void PrintModelInfo(ge::proto::ModelDef *model_def) { | FMK_FUNC_HOST_VISIBILITY void PrintModelInfo(ge::proto::ModelDef *model_def) { | ||||
| std::cout << "------------ Display Model Info start ------------" << std::endl; | std::cout << "------------ Display Model Info start ------------" << std::endl; | ||||
| auto model_attr_map = model_def->mutable_attr(); | |||||
| // system info | // system info | ||||
| iter = model_attr_map->find(ATTR_MODEL_ATC_VERSION); | 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() : ""; | ||||
| @@ -888,7 +889,6 @@ FMK_FUNC_HOST_VISIBILITY void PrintModelInfo(ge::proto::ModelDef *model_def) { | |||||
| << "[" << framework_type << "]." << std::endl; | << "[" << framework_type << "]." << std::endl; | ||||
| // resource info | // resource info | ||||
| auto model_attr_map = model_def->mutable_attr(); | |||||
| auto iter = model_attr_map->find(ATTR_MODEL_MEMORY_SIZE); | auto 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); | ||||