Fix GroupConvolutionTranspose OP clone function.

17fe19ea · Sang Ik Lee · abdd9c0b · 17fe19ea
Commit 17fe19ea authored Jun 28, 2019 by Sang Ik Lee
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

group_conv_transpose.cpp src/ngraph/op/fused/group_conv_transpose.cpp +2 -1

No files found.
--- a/src/ngraph/op/fused/group_conv_transpose.cpp
+++ b/src/ngraph/op/fused/group_conv_transpose.cpp
@@ -203,7 +203,8 @@ shared_ptr<Node> op::GroupConvolutionTranspose::copy_with_new_args(const NodeVec
                                                      get_padding_end(),
                                                      get_output_padding(),
                                                      get_groups(),
-                                                      get_pad_type());
+                                                      get_pad_type(),
+                                                      get_output_shape());
 }

 Shape op::GroupConvolutionTranspose::get_data_batch_shape() const