pull/4408/head
yankai 5 years ago
parent 8c377fd159
commit 3dc3b8153e

@ -35,4 +35,5 @@ int AnfMatmulPopulater::Populate(const PrimitivePtr &prim, PrimitiveTValue *prim
return 0; return 0;
} }
AnfNodePopulaterRegistrar anfMatmulPopulater("Matmul", new AnfMatmulPopulater()); AnfNodePopulaterRegistrar anfMatmulPopulater("Matmul", new AnfMatmulPopulater());
AnfNodePopulaterRegistrar anfMatMulPopulater("MatMul", new AnfMatmulPopulater());
} // namespace mindspore::lite } // namespace mindspore::lite

@ -25,8 +25,8 @@ int AnfReshapePopulater::Populate(const PrimitivePtr &prim, PrimitiveTValue *pri
const std::vector<AnfNodePtr> &inputs) { const std::vector<AnfNodePtr> &inputs) {
auto primitive = std::make_unique<schema::PrimitiveT>(); auto primitive = std::make_unique<schema::PrimitiveT>();
auto attr = std::make_unique<schema::ReshapeT>(); auto attr = std::make_unique<schema::ReshapeT>();
MS_ASSERT(inputs.size() == kAnfPopulaterThree); MS_ASSERT(inputs.size() == kAnfPopulaterThree - 1);
auto inputNode = inputs[kAnfPopulaterTwo]; auto inputNode = inputs[kAnfPopulaterTwo - 1];
if (inputNode->isa<ValueNode>()) { if (inputNode->isa<ValueNode>()) {
auto valueNode = inputNode->cast<ValueNodePtr>(); auto valueNode = inputNode->cast<ValueNodePtr>();
MS_ASSERT(valueNode != nullptr); MS_ASSERT(valueNode != nullptr);

Loading…
Cancel
Save