Browse Source

fix_return_scalar

tags/v1.1.0
jjfeing 5 years ago
parent
commit
a607890256
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      mindspore/ccsrc/backend/optimizer/pass/convert_const_scalar_to_tensor.cc

+ 4
- 0
mindspore/ccsrc/backend/optimizer/pass/convert_const_scalar_to_tensor.cc View File

@@ -60,6 +60,10 @@ const AnfNodePtr ConvertConstScalarToTensor::Process(const FuncGraphPtr &func_gr
if (node == nullptr || func_graph == nullptr || AnfAlgo::CheckPrimitiveType(node, prim::kPrimTupleGetItem)) { if (node == nullptr || func_graph == nullptr || AnfAlgo::CheckPrimitiveType(node, prim::kPrimTupleGetItem)) {
return nullptr; 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>()) { if (!node->isa<CNode>()) {
return nullptr; return nullptr;
} }


Loading…
Cancel
Save