diff --git a/mindspore/lite/src/model_impl.cc b/mindspore/lite/src/model_impl.cc index 25273b928f..7854161fab 100644 --- a/mindspore/lite/src/model_impl.cc +++ b/mindspore/lite/src/model_impl.cc @@ -227,6 +227,12 @@ lite::Primitive *ModelImpl::CopyPrimitive(const schema::Primitive *src_prim) { return new lite::Tile(const_cast(src_prim)); case schema::PrimitiveType_Resize: return new lite::Resize(const_cast(src_prim)); + case schema::PrimitiveType_Unstack: + return new lite::Unstack(const_cast(src_prim)); + case schema::PrimitiveType_Unique: + return new lite::Unique(const_cast(src_prim)); + case schema::PrimitiveType_TopK: + return new lite::TopK(const_cast(src_prim)); case schema::PrimitiveType_MatMul: return new lite::MatMul(const_cast(src_prim)); case schema::PrimitiveType_QuantDTypeCast: