!8921 fix return scalar

From: @jjfeing
Reviewed-by: @kisnwang,@chujinjin
Signed-off-by: @chujinjin
pull/8921/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 661b6073a4

@ -60,6 +60,10 @@ const AnfNodePtr ConvertConstScalarToTensor::Process(const FuncGraphPtr &func_gr
if (node == nullptr || func_graph == nullptr || AnfAlgo::CheckPrimitiveType(node, prim::kPrimTupleGetItem)) {
return nullptr;
}
// input is scalar, and link to graph return
if (node->isa<ValueNode>() && node == func_graph->output()) {
return CreateTensorInput(func_graph->cast<KernelGraphPtr>(), node);
}
if (!node->isa<CNode>()) {
return nullptr;
}

Loading…
Cancel
Save