From 3bd69c39617c535043fae6a9b92ad4325e264adf Mon Sep 17 00:00:00 2001 From: AGroupofProbiotocs Date: Fri, 18 Sep 2020 15:24:47 +0800 Subject: [PATCH] add available preset MindSpore model - shufflenetv2 --- mindspore/lite/src/ops/primitive_c.cc | 3 +++ mindspore/lite/test/models_mindspore.cfg | 3 ++- 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/mindspore/lite/src/ops/primitive_c.cc b/mindspore/lite/src/ops/primitive_c.cc index 29f101416a..7b4f8ce457 100644 --- a/mindspore/lite/src/ops/primitive_c.cc +++ b/mindspore/lite/src/ops/primitive_c.cc @@ -446,6 +446,9 @@ std::shared_ptr PrimitiveC::Create(const Primitive &prim, const std: return NewPrimitiveC(prim, inputs, quantType); } else if (op_type == "Tile") { return NewPrimitiveC(prim, inputs, quantType); +#else + } else if (op_type == "Conv2DBackpropInput") { + return NewPrimitiveC(prim, inputs, quantType); #endif } else { MS_LOG(ERROR) << "Unsupported primitive type in Create : " << op_type; diff --git a/mindspore/lite/test/models_mindspore.cfg b/mindspore/lite/test/models_mindspore.cfg index a41c9b1ba6..0ddc80cbaa 100644 --- a/mindspore/lite/test/models_mindspore.cfg +++ b/mindspore/lite/test/models_mindspore.cfg @@ -1,3 +1,4 @@ ssd.mindir mobilenetv2_438.mindir -gate_u_net_small-1_110.mindir \ No newline at end of file +gate_u_net_small-1_110.mindir +shufflenetv2.mindir