diff --git a/paddle/fluid/operators/conv_transpose_op.cc b/paddle/fluid/operators/conv_transpose_op.cc index a4f00f6cd8..6803622c12 100644 --- a/paddle/fluid/operators/conv_transpose_op.cc +++ b/paddle/fluid/operators/conv_transpose_op.cc @@ -662,3 +662,36 @@ REGISTER_OP_VERSION(conv_transpose) "In order to add additional size to one side of each dimension " "in the output", {})); + +REGISTER_OP_VERSION(conv2d_transpose) + .AddCheckpoint( + R"ROC( + Upgrade conv2d transpose to add a new attribute [output_padding]. + )ROC", + paddle::framework::compatible::OpVersionDesc().NewAttr( + "output_padding", + "In order to add additional size to one side of each dimension " + "in the output", + {})); + +REGISTER_OP_VERSION(conv3d_transpose) + .AddCheckpoint( + R"ROC( + Upgrade conv3d transpose to add a new attribute [output_padding]. + )ROC", + paddle::framework::compatible::OpVersionDesc().NewAttr( + "output_padding", + "In order to add additional size to one side of each dimension " + "in the output", + {})); + +REGISTER_OP_VERSION(depthwise_conv2d_transpose) + .AddCheckpoint( + R"ROC( + Upgrade depthwise conv2d transpose to add a new attribute [output_padding]. + )ROC", + paddle::framework::compatible::OpVersionDesc().NewAttr( + "output_padding", + "In order to add additional size to one side of each dimension " + "in the output", + {}));