Commit 40b39a1b authored by Adam Procter's avatar Adam Procter Committed by Scott Cyphers

Preserve pad_mode in reshape sinking for Pad (#3209)

parent 2427b205
......@@ -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);
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment