From ea302afb4713594e1b1c21d7021dc341fdc60ad7 Mon Sep 17 00:00:00 2001
From: Andrew Kwangwoong Park <andrew.park@intel.com>
Date: Wed, 10 Aug 2022 10:37:51 +0900
Subject: [PATCH] Update pre_replace_deconv to support output_shape for
 transposed conv (#12418)

Signed-off-by: Andrew Park <andrew.park@intel.com>
---
 .../intel_gpu/primitives/convolution.hpp      |  4 +-
 .../graph_optimizer/pre_replace_deconv.cpp    | 37 ++++++++++++++-----
 2 files changed, 30 insertions(+), 11 deletions(-)

diff --git a/src/plugins/intel_gpu/include/intel_gpu/primitives/convolution.hpp b/src/plugins/intel_gpu/include/intel_gpu/primitives/convolution.hpp
index ab3073ab7e2..be0ce3037e8 100644
--- a/src/plugins/intel_gpu/include/intel_gpu/primitives/convolution.hpp
+++ b/src/plugins/intel_gpu/include/intel_gpu/primitives/convolution.hpp
@@ -511,6 +511,7 @@ struct convolution : public primitive_base<convolution> {
                 ov::Strides stride = {1, 1},
                 ov::CoordinateDiff pad = {0, 0},
                 ov::Strides dilation = {1, 1},
+                tensor output_size = {0, 0, 0, 0},
                 bool grouped_weights_shape = false,
                 const primitive_id& ext_prim_id = "",
                 const padding& output_padding = padding())
@@ -518,7 +519,8 @@ struct convolution : public primitive_base<convolution> {
           pad(pad),
           stride(stride),
           dilation(dilation),
-          with_output_size(false),
+          with_output_size(output_size.count() > 0 ? true : false),
+          output_size(output_size),
           groups(groups),
           deformable_groups(1),
           padding_above(stride.size(), 0),
diff --git a/src/plugins/intel_gpu/src/graph/graph_optimizer/pre_replace_deconv.cpp b/src/plugins/intel_gpu/src/graph/graph_optimizer/pre_replace_deconv.cpp
index be8a4b1ca08..40b7f195199 100644
--- a/src/plugins/intel_gpu/src/graph/graph_optimizer/pre_replace_deconv.cpp
+++ b/src/plugins/intel_gpu/src/graph/graph_optimizer/pre_replace_deconv.cpp
@@ -121,16 +121,33 @@ void pre_replace_deconv::run(program& p) {
                                                               "",
                                                               output_padding);
                 } else {
-                    conv_prim = std::make_shared<convolution>(deconv_node_id,
-                                                              input_node_id,
-                                                              weights_nodes_id,
-                                                              groups,
-                                                              stride,
-                                                              pad,
-                                                              dilation,
-                                                              grouped_weights_shape,
-                                                              "",
-                                                              output_padding);
+                    tensor output_size(0);
+                    if (deconv_prim->with_output_size) {
+                        output_size = deconv_prim->output_size;
+                        conv_prim = std::make_shared<convolution>(deconv_node_id,
+                                                                  input_node_id,
+                                                                  weights_nodes_id,
+                                                                  groups,
+                                                                  stride,
+                                                                  pad,
+                                                                  dilation,
+                                                                  output_size,
+                                                                  grouped_weights_shape,
+                                                                  "",
+                                                                  output_padding);
+                    } else {
+                        conv_prim = std::make_shared<convolution>(deconv_node_id,
+                                                                  input_node_id,
+                                                                  weights_nodes_id,
+                                                                  groups,
+                                                                  stride,
+                                                                  pad,
+                                                                  dilation,
+                                                                  output_size,
+                                                                  grouped_weights_shape,
+                                                                  "",
+                                                                  output_padding);
+                    }
                 }
                 program_node& new_node = p.get_or_create(conv_prim);