diff --git a/paddle/fluid/framework/ir/delete_repeated_ops_pass.cc b/paddle/fluid/framework/ir/delete_repeated_ops_pass.cc index cf5bb15c207e5d00c7b0f423752e5aa26509ca99..2f85d5b288025b437a7c594f279c78e344c97dbb 100644 --- a/paddle/fluid/framework/ir/delete_repeated_ops_pass.cc +++ b/paddle/fluid/framework/ir/delete_repeated_ops_pass.cc @@ -151,6 +151,8 @@ std::string GenSliceAttrKey(OpDesc* slice_op_desc) { auto starts = slice_op_desc->GetAttrIfExists>("starts"); auto ends = slice_op_desc->GetAttrIfExists>("ends"); auto axes = slice_op_desc->GetAttrIfExists>("axes"); + auto decrease_axis = + slice_op_desc->GetAttrIfExists>("decrease_axis"); attr_key += "starts_"; for (auto start : starts) { attr_key += std::to_string(start) + "_"; @@ -163,6 +165,10 @@ std::string GenSliceAttrKey(OpDesc* slice_op_desc) { for (auto axis : axes) { attr_key += std::to_string(axis) + "_"; } + attr_key += "decrease_axis_"; + for (auto axis : decrease_axis) { + attr_key += std::to_string(axis) + "_"; + } return attr_key; }