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

!1099 fix bug of can not find op adapter 

Merge pull request !1099 from fary86/fix_bug_for_can_not_find_opadapter
This commit is contained in:
mindspore-ci-bot 2020-05-13 11:10:32 +08:00 committed by Gitee
parent 168dfb2555 77ef790f25
commit a2d5ad5abe
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/ccsrc/transform/convert.cc
View File

@ -429,9 +429,8 @@ OpAdapterPtr DfGraphConvertor::FindAdapter(const AnfNodePtr node, bool train) {
auto it_adpt = get_adpt_map().find(name);
if (it_adpt != get_adpt_map().end()) {
return it_adpt->second->Get(train);
} else {
MS_LOG(ERROR) << "Can't find OpAdapter for " << name;
}
MS_LOG(EXCEPTION) << "Can't find OpAdapter for " << name;
}
if (node->isa<ValueNode>()) {
@ -521,8 +520,7 @@ OpAdapterPtr DfGraphConvertor::FindAdapter(const std::string &name, bool train)
if (it != get_adpt_map().end()) {
return it->second->Get(train);
}
MS_LOG(ERROR) << "Can't find OpAdapter for " << name;
return transform::OpAdapterPtr(nullptr);
MS_LOG(EXCEPTION) << "Can't find OpAdapter for " << name;
}
void DfGraphConvertor::DrawParamInitSubGraph(const std::string &name, const AnfNodePtr &it) {