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