From 3dc3b8153e168d89e1e89edeaf6c1ff89ff6ec68 Mon Sep 17 00:00:00 2001 From: yankai Date: Thu, 13 Aug 2020 20:10:52 +0800 Subject: [PATCH] fix bug --- .../common/anf_importer/anf_populater/anf_matmul_populater.cc | 1 + .../anf_importer/anf_populater/anf_reshape_populater.cc | 4 ++-- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/mindspore/lite/src/common/anf_importer/anf_populater/anf_matmul_populater.cc b/mindspore/lite/src/common/anf_importer/anf_populater/anf_matmul_populater.cc index 7b5b5bb4d9..109f7dea7a 100644 --- a/mindspore/lite/src/common/anf_importer/anf_populater/anf_matmul_populater.cc +++ b/mindspore/lite/src/common/anf_importer/anf_populater/anf_matmul_populater.cc @@ -35,4 +35,5 @@ int AnfMatmulPopulater::Populate(const PrimitivePtr &prim, PrimitiveTValue *prim return 0; } AnfNodePopulaterRegistrar anfMatmulPopulater("Matmul", new AnfMatmulPopulater()); +AnfNodePopulaterRegistrar anfMatMulPopulater("MatMul", new AnfMatmulPopulater()); } // namespace mindspore::lite diff --git a/mindspore/lite/src/common/anf_importer/anf_populater/anf_reshape_populater.cc b/mindspore/lite/src/common/anf_importer/anf_populater/anf_reshape_populater.cc index 9baea4130b..6695faaae6 100644 --- a/mindspore/lite/src/common/anf_importer/anf_populater/anf_reshape_populater.cc +++ b/mindspore/lite/src/common/anf_importer/anf_populater/anf_reshape_populater.cc @@ -25,8 +25,8 @@ int AnfReshapePopulater::Populate(const PrimitivePtr &prim, PrimitiveTValue *pri const std::vector &inputs) { auto primitive = std::make_unique(); auto attr = std::make_unique(); - MS_ASSERT(inputs.size() == kAnfPopulaterThree); - auto inputNode = inputs[kAnfPopulaterTwo]; + MS_ASSERT(inputs.size() == kAnfPopulaterThree - 1); + auto inputNode = inputs[kAnfPopulaterTwo - 1]; if (inputNode->isa()) { auto valueNode = inputNode->cast(); MS_ASSERT(valueNode != nullptr);