Preserve pad_mode in reshape sinking for Pad (#3209)

NervanaSystems · Jul 11, 2019 · 40b39a1 · 40b39a1
1 parent 2427b20
commit 40b39a1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/ngraph/pass/reshape_sinking.cpp b/src/ngraph/pass/reshape_sinking.cpp
@@ -376,8 +376,8 @@ static void
 
     auto new_lower = ngraph::apply_permutation(n->get_padding_below(), def_order);
     auto new_upper = ngraph::apply_permutation(n->get_padding_above(), def_order);
-    auto new_pad =
-        make_shared<op::Pad>(dummy_correct_shape, n->get_argument(1), new_lower, new_upper);
+    auto new_pad = make_shared<op::Pad>(
+        dummy_correct_shape, n->get_argument(1), new_lower, new_upper, n->get_pad_mode());
     ngraph::replace_node(dummy_correct_shape, n->get_argument(0));
     NGRAPH_DEBUG << "Replacing " << n->get_name() << " with " << new_pad->get_name();
     ngraph::replace_node(n, new_pad);