|
|
|
@@ -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 |