Browse Source

display model info

tags/v1.2.0
wangwenhua1@huawei.com 3 years ago
parent
commit
8617dcc304
1 changed files with 2 additions and 3 deletions
  1. +2
    -3
      tests/ut/ge/session/omg_omg_unittest.cc

+ 2
- 3
tests/ut/ge/session/omg_omg_unittest.cc View File

@@ -33,7 +33,7 @@ class UtestOmg : public testing::Test {
TEST_F(UtestOmg, display_model_info_failed) { TEST_F(UtestOmg, display_model_info_failed) {
ge::proto::ModelDef model_def; ge::proto::ModelDef model_def;
PrintModelInfo(&model_def);
PrintModelInfo(&model_def, 1);
} }
TEST_F(UtestOmg, display_model_info_success) { TEST_F(UtestOmg, display_model_info_success) {
@@ -46,7 +46,6 @@ TEST_F(UtestOmg, display_model_info_success) {
attr_def->mutable_list()->add_i(2); attr_def->mutable_list()->add_i(2);
attr_def->mutable_list()->add_i(3); attr_def->mutable_list()->add_i(3);
attr_def->mutable_list()->add_i(4); attr_def->mutable_list()->add_i(4);
attr_def->mutable_list()->add_i(5);
PrintModelInfo(&model_def);
PrintModelInfo(&model_def, 1);
} }
} // namespace ge } // namespace ge

Loading…
Cancel
Save