From a79311fa1ac5ee58b3554e0aec0d8a19d40522de Mon Sep 17 00:00:00 2001 From: Margaret_wangrui Date: Thu, 28 May 2020 17:59:18 +0800 Subject: [PATCH] handle tuple in CreateNewCNode --- mindspore/ccsrc/session/session_basic.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mindspore/ccsrc/session/session_basic.cc b/mindspore/ccsrc/session/session_basic.cc index 928ac88501..a3f1ccbbb3 100644 --- a/mindspore/ccsrc/session/session_basic.cc +++ b/mindspore/ccsrc/session/session_basic.cc @@ -375,7 +375,7 @@ CNodePtr SessionBasic::CreateNewCNode(const CNodePtr &cnode, bool valid_input, K cnode_inputs.emplace_back(new_value_node); } continue; - } else if (anf->isa()) { + } else if (anf->isa() && AnfAlgo::GetOutputTensorNum(anf) == 1) { auto new_parameter = CreateNewParameterFromParameter(anf, valid_input, graph); cnode_inputs.push_back(new_parameter); if (GetGraphIdByNode(anf) == kInvalidGraphId) { @@ -384,7 +384,7 @@ CNodePtr SessionBasic::CreateNewCNode(const CNodePtr &cnode, bool valid_input, K (*other_graph_cnode)[anf] = new_parameter; } continue; - } else if (anf->isa()) { + } else if (anf->isa()) { *from_other_graph = true; // the input node is a cnode from other graph auto parameter_from_cnode = CreateNewParameterFromCNode(anf, valid_input, graph);