未验证 提交 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 { ...@@ -37,8 +37,13 @@ class InplaceOpVarPass : public FusePassBase {
std::vector<std::string> GetControlFlowVarNames(ir::Graph* graph) const; std::vector<std::string> GetControlFlowVarNames(ir::Graph* graph) const;
std::set<std::string> inplace_ops_{ std::set<std::string> inplace_ops_{"reshape",
"reshape", "unsqueeze", "unsqueeze2", "squeeze", "squeeze2"}; "reshape2",
"unsqueeze",
"unsqueeze2",
"squeeze",
"squeeze2",
"flatten_contiguous_range"};
std::set<std::string> control_flow_ops_{"while", "conditional_block"}; 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.
先完成此消息的编辑!
想要评论请 注册