Browse Source

dynamic shape inference support

pull/1231/head
lichun 4 years ago
parent
commit
28f2a8686d
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      tests/ut/ge/generator/ge_generator_unittest.cc

+ 2
- 1
tests/ut/ge/generator/ge_generator_unittest.cc View File

@@ -77,6 +77,7 @@ TEST_F(UtestGeGenerator, test_generate_model) {
string file_name_prefix = "123";
const vector<GeTensor> inputs = {};
ModelBufferData model_data;
EXPECT_EQ(GenerateModel(graph, file_name_prefix, inputs, model_data), FAILED);
GeGenerator generator;
EXPECT_EQ(generator.GenerateModel(graph, file_name_prefix, inputs, model_data, true), FAILED);
}
} // namespace ge

Loading…
Cancel
Save