From e6b0ce04a1486848078cd1729fccce9a816a5ec8 Mon Sep 17 00:00:00 2001 From: Abhishek Kulkarni Date: Tue, 1 Dec 2020 22:44:13 -0800 Subject: [PATCH] Preserve node "friendly" names in the transpose sinking pass Use replace_node_update_name to preserve friendly names of replaced nodes in the transpose sinking pass. --- ngraph_bridge/pass/transpose_sinking.cc | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/ngraph_bridge/pass/transpose_sinking.cc b/ngraph_bridge/pass/transpose_sinking.cc index e5ba54984..e1b13b226 100644 --- a/ngraph_bridge/pass/transpose_sinking.cc +++ b/ngraph_bridge/pass/transpose_sinking.cc @@ -267,7 +267,7 @@ static void sink_transpose( // should be safe to remove an already detached node mark_transpose_for_deletion(orig_transpose, transposes_to_delete); // replace transpose with combined one - ngraph::replace_node(transpose, new_transpose); + ngraph::replace_node_update_name(transpose, new_transpose); mark_transpose_for_deletion(new_transpose, transposes_to_delete); write_transposemap(reorders, new_transpose, new_transpose); } @@ -361,11 +361,11 @@ static void sink_pad( auto new_pad = make_shared(dummy_correct_shape, new_begin, new_end, n->input_value(3), n->get_pad_mode()); - ngraph::replace_node(dummy_correct_shape, - n->input_value(0).get_node_shared_ptr()); + ngraph::replace_node_update_name(dummy_correct_shape, + n->input_value(0).get_node_shared_ptr()); NGRAPH_VLOG(4) << "Replacing " << n->get_name() << " with " << new_pad->get_name(); - ngraph::replace_node(n, new_pad); + ngraph::replace_node_update_name(n, new_pad); auto new_transpose = make_transpose(new_pad, order); NGRAPH_VLOG(4) << "Propagating " << describe(new_transpose) << " for " << n->get_name(); @@ -421,7 +421,7 @@ static void sink_concat(shared_ptr n, TransposeMap& reorders, } NGRAPH_VLOG(4) << "Replacing " << n->get_name() << " with " << new_concat->get_name(); - ngraph::replace_node(n, new_concat); + ngraph::replace_node_update_name(n, new_concat); auto new_transpose = make_transpose(new_concat, order); NGRAPH_VLOG(4) << "Propagating " << describe(new_transpose) << " for " << n->get_name();