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);