From be8b5fd18a99014e36ee24c8cafd2802d92ffd06 Mon Sep 17 00:00:00 2001
From: wangxinxin08 <69842442+wangxinxin08@users.noreply.github.com>
Date: Tue, 29 Dec 2020 14:43:22 +0800
Subject: [PATCH] register op version for conv2d_transpose, conv3d_transpose
 and depthwise_conv2d_transpose, test=op_version (#29937)

---
 paddle/fluid/operators/conv_transpose_op.cc | 33 +++++++++++++++++++++
 1 file changed, 33 insertions(+)

diff --git a/paddle/fluid/operators/conv_transpose_op.cc b/paddle/fluid/operators/conv_transpose_op.cc
index a4f00f6cd80..6803622c128 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",
+            {}));
-- 
GitLab