未验证 提交 dd0681e3 编写于 作者: C chen 提交者: GitHub

[IR] Reshape2 and Flatten_contiguous_range Support Inplace (#55809)

* inplace pass support reshape2 and flatten_contiguous_range

* recover the modification to inplace_op_var_pass.cc
上级 bd14926e
......@@ -37,8 +37,13 @@ class InplaceOpVarPass : public FusePassBase {
std::vector<std::string> GetControlFlowVarNames(ir::Graph* graph) const;
std::set<std::string> inplace_ops_{
"reshape", "unsqueeze", "unsqueeze2", "squeeze", "squeeze2"};
std::set<std::string> inplace_ops_{"reshape",
"reshape2",
"unsqueeze",
"unsqueeze2",
"squeeze",
"squeeze2",
"flatten_contiguous_range"};
std::set<std::string> control_flow_ops_{"while", "conditional_block"};
};
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册