|
|
@ -226,6 +226,8 @@ lite::Primitive *ModelImpl::CopyPrimitive(const schema::Primitive *srcPrim) {
|
|
|
|
return new lite::EmbeddingLookup(const_cast<schema::Primitive *>(srcPrim));
|
|
|
|
return new lite::EmbeddingLookup(const_cast<schema::Primitive *>(srcPrim));
|
|
|
|
case schema::PrimitiveType_Elu:
|
|
|
|
case schema::PrimitiveType_Elu:
|
|
|
|
return new lite::Elu(const_cast<schema::Primitive *>(srcPrim));
|
|
|
|
return new lite::Elu(const_cast<schema::Primitive *>(srcPrim));
|
|
|
|
|
|
|
|
case schema::PrimitiveType_DeDepthwiseConv2D:
|
|
|
|
|
|
|
|
return new lite::DeconvDepthwiseConv2D(const_cast<schema::Primitive *>(srcPrim));
|
|
|
|
default:
|
|
|
|
default:
|
|
|
|
break;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|