xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!22976 [MS][Lite]Model converted wrong with if_while example. 

Merge pull request !22976 from mengyuanli/bugfix_mindIR
This commit is contained in:
i-robot 2021-09-08 09:13:09 +00:00 committed by Gitee
parent 4cab7ffbd3 e6b2933150
commit ff102f8619
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/lite/tools/converter/import/mindir_control_flow_adjust.cc
View File

@ -68,7 +68,7 @@ std::vector<AnfNodePtr> MindIRControlFlowAdjust::GetFgOutput(const FuncGraphPtr
auto output_cnode = output_node->cast<CNodePtr>();
if (output_cnode == nullptr) {
MS_LOG(INFO) << "graph output is not cnode.";
return {};
return {output_node};
}
if (!IsMakeTuple(output_node)) {
MS_LOG(INFO) << "graph is single output.";