diff --git a/paddle/fluid/framework/ir/inplace_op_var_pass.h b/paddle/fluid/framework/ir/inplace_op_var_pass.h index 50c9c502915316ee4730e309fe30f14c0105983b..a06e266c38f4b7a2e7f510824be624ee6d1612b6 100644 --- a/paddle/fluid/framework/ir/inplace_op_var_pass.h +++ b/paddle/fluid/framework/ir/inplace_op_var_pass.h @@ -37,8 +37,13 @@ class InplaceOpVarPass : public FusePassBase { std::vector GetControlFlowVarNames(ir::Graph* graph) const; - std::set inplace_ops_{ - "reshape", "unsqueeze", "unsqueeze2", "squeeze", "squeeze2"}; + std::set inplace_ops_{"reshape", + "reshape2", + "unsqueeze", + "unsqueeze2", + "squeeze", + "squeeze2", + "flatten_contiguous_range"}; std::set control_flow_ops_{"while", "conditional_block"}; };