提交 b348e159 编写于 作者: Y Yancey1989

fix rename var

上级 2013b1cc
...@@ -243,13 +243,8 @@ const std::unordered_map<std::string, Attribute> &OpDesc::GetAttrMap() const { ...@@ -243,13 +243,8 @@ const std::unordered_map<std::string, Attribute> &OpDesc::GetAttrMap() const {
} }
void OpDesc::Rename(const std::string &old_name, const std::string &new_name) { void OpDesc::Rename(const std::string &old_name, const std::string &new_name) {
for (auto &input : inputs_) { RenameInput(old_name, new_name);
std::replace(input.second.begin(), input.second.end(), old_name, new_name); RenameOutput(old_name, new_name);
}
for (auto &output : outputs_) {
std::replace(output.second.begin(), output.second.end(), old_name,
new_name);
}
need_update_ = true; need_update_ = true;
} }
...@@ -274,6 +269,13 @@ void OpDesc::RenameInput(const std::string &old_name, ...@@ -274,6 +269,13 @@ void OpDesc::RenameInput(const std::string &old_name,
for (auto &input : inputs_) { for (auto &input : inputs_) {
std::replace(input.second.begin(), input.second.end(), old_name, new_name); std::replace(input.second.begin(), input.second.end(), old_name, new_name);
} }
auto it = attrs_.find(framework::OpProtoAndCheckerMaker::OpRoleVarAttrName());
if (it != attrs_.end()) {
auto &op_vars = boost::get<std::vector<std::string>>(it->second);
std::replace(op_vars.begin(), op_vars.end(), old_name, new_name);
}
need_update_ = true; need_update_ = true;
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册