|
|
@@ -128,4 +128,13 @@ TEST_F(UtestGeGenerator, test_set_model_name) { |
|
|
|
ge_root_model->root_graph_ = std::move(graph); |
|
|
|
EXPECT_EQ(generator.SetModelNameForDump(ge_root_model), SUCCESS); |
|
|
|
} |
|
|
|
|
|
|
|
TEST_F(UtestGeGenerator, test_remove_const) { |
|
|
|
GeGenerator generator; |
|
|
|
GeTensorDesc tensor_desc; |
|
|
|
GeTensor tensor(tensor_desc); |
|
|
|
const vector<GeTensor> inputs = {tensor}; |
|
|
|
vector<GeTensor> outputs; |
|
|
|
generator.RemoveConst(inputs, outputs); |
|
|
|
} |
|
|
|
} // namespace ge |