|
|
|
@@ -910,6 +910,12 @@ FMK_FUNC_HOST_VISIBILITY void PrintModelInfo(ge::proto::ModelDef *model_def) { |
|
|
|
|
|
|
|
// om info |
|
|
|
iter = model_attr_map->find("om_info_list"); |
|
|
|
if (iter == model_attr_map->end()) { |
|
|
|
std::cout << "Display Model Info error, om_info_list is not find in om, please check atc version is matched." |
|
|
|
<< std::endl; |
|
|
|
std::cout << "------------ Display Model Info end ------------" << std::endl; |
|
|
|
return; |
|
|
|
} |
|
|
|
auto bt = iter->second.list().i_size(); |
|
|
|
if (bt == 5) { |
|
|
|
std::cout << "om info: " |
|
|
|
|