pull/1311/head
zhaozhixuan 4 years ago
parent e1eb148756
commit 6af2a247fd

@ -714,7 +714,7 @@ void GeGenerator::RemoveConst(const vector<GeTensor> &inputs, vector<GeTensor> &
for (auto input : inputs) {
GeTensorDesc input_desc = input.GetTensorDesc();
bool is_const = false;
(void)AttrUtils::GetBool(tensor, CONST_ATTR_NAME_INPUT, is_const);
(void)AttrUtils::GetBool(input_desc, CONST_ATTR_NAME_INPUT, is_const);
if (!is_const) {
outputs.emplace_back(input);
}

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

Loading…
Cancel
Save