display model info

ge/generator/ge_generator.cc

@@ -928,7 +928,7 @@ Status GeGenerator::Impl::SaveRootModel(const string &file_name_prefix, GeRootMo
     GELOGE(ret, "Save to om model failed");
     return ret;
   }
-  DisplayModelInfo(*(model_root.get()));
+  // DisplayModelInfo(*(model_root.get()));
   return SUCCESS;
 }
 

ge/offline/main.cc

@@ -1180,6 +1180,11 @@ domi::Status GenerateOmModel() {
     return domi::FAILED;
   }
 
+  if (FLAGS_display_model_info == "1") {
+    GELOGI("need to display model info.");
+    return ge::ConvertOm(FLAGS_output.c_str(), "", false);
+  }
+
   return domi::SUCCESS;
 }