From 5b7a0a59a1f9a0da1fd103eb380fc19b8cbad2de Mon Sep 17 00:00:00 2001 From: sunsuodong Date: Sat, 15 Aug 2020 15:05:35 +0800 Subject: [PATCH] fix_copy_primitive --- mindspore/lite/src/model_impl.cc | 6 ++++++ 1 file changed, 6 insertions(+) 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: