Unverified Commit e8735112 authored by Nick Korovaiko's avatar Nick Korovaiko Committed by GitHub

Update reshape_sinking.cpp

parent a24e9654
......@@ -436,7 +436,7 @@ static void sink_concat(std::shared_ptr<op::Concat> n,
auto iinput_shape = ngraph::apply_permutation(iarg_reshape->get_shape(), def_order);
auto idummy_correct_shape =
std::make_shared<pattern::op::Label>(iarg_reshape->get_element_type(), input_shape);
std::make_shared<pattern::op::Label>(iarg_reshape->get_element_type(), iinput_shape);
new_args.push_back(idummy_correct_shape);
}
......
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