p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!5573 Fix wrong graph inputs bug 

Merge pull request !5573 from fanglei/graphclone
This commit is contained in:
mindspore-ci-bot 2020-09-01 10:45:21 +08:00 committed by Gitee
parent 1d0e0ae27c d69224f569
commit 340bb4c28f
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mindspore/core/ir/func_graph_cloner.cc
View File

@ -336,7 +336,14 @@ void Cloner::AddInputs(const FuncGraphPtr &func_graph_user, const FuncGraphPtr &
}
auto cnode = node->cast<CNodePtr>();
auto inputs = cnode->inputs();
(void)std::copy(params.begin(), params.end(), std::back_inserter(inputs));
std::vector<AnfNodePtr> add_params{params.begin(), params.end()};
for (size_t i = 2; i < inputs.size(); i++) {
auto ret = std::find(add_params.begin(), add_params.end(), inputs[i]);
if (ret != add_params.end()) {
add_params.erase(ret);
}
}
(void)std::copy(add_params.begin(), add_params.end(), std::back_inserter(inputs));
cnode->set_inputs(inputs);
OrderParameters(func_graph, inputs);
}