|
|
|
@ -273,10 +273,11 @@ int DeConv2D::UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffer
|
|
|
|
|
MS_LOG(ERROR) << "value_as_DeConv2D return nullptr";
|
|
|
|
|
return RET_ERROR;
|
|
|
|
|
}
|
|
|
|
|
auto val_offset = schema::CreateDeConv2D(
|
|
|
|
|
*fbb, attr->format(), attr->group(), attr->channelIn(), attr->channelOut(), attr->kernelW(), attr->kernelH(),
|
|
|
|
|
attr->strideW(), attr->strideH(), attr->padMode(), attr->padUp(), attr->padDown(), attr->padLeft(),
|
|
|
|
|
attr->padRight(), attr->dilateW(), attr->dilateH(), attr->hasBias(), attr->activationType());
|
|
|
|
|
auto val_offset =
|
|
|
|
|
schema::CreateDeConv2D(*fbb, attr->format(), attr->group(), attr->channelIn(), attr->channelOut(), attr->kernelW(),
|
|
|
|
|
attr->kernelH(), attr->strideW(), attr->strideH(), attr->padMode(), attr->padUp(),
|
|
|
|
|
attr->padDown(), attr->padLeft(), attr->padRight(), attr->dilateW(), attr->dilateH(),
|
|
|
|
|
attr->hasBias(), attr->activationType(), attr->outputPaddingW(), attr->outputPaddingH());
|
|
|
|
|
auto prim_offset = schema::CreatePrimitive(*fbb, schema::PrimitiveType_DeConv2D, val_offset.o);
|
|
|
|
|
fbb->Finish(prim_offset);
|
|
|
|
|
return RET_OK;
|
|
|
|
|