Commit 57f4f44b authored by Nick Korovaiko's avatar Nick Korovaiko Committed by Robert Kimball

broadcast fix (#1483)

parent ebd6904c
...@@ -109,7 +109,7 @@ bool runtime::cpu::pass::CPUCollapseDims::run_on_function(std::shared_ptr<ngraph ...@@ -109,7 +109,7 @@ bool runtime::cpu::pass::CPUCollapseDims::run_on_function(std::shared_ptr<ngraph
// Null broadcast operation, replace with reshape // Null broadcast operation, replace with reshape
AxisVector axis_order = ngraph::get_default_order(input_shape); AxisVector axis_order = ngraph::get_default_order(input_shape);
auto reshape = std::make_shared<op::Reshape>( auto reshape = std::make_shared<op::Reshape>(
node->get_argument(0), axis_order, Shape(cdims.output_shape)); node->get_argument(0), axis_order, n->get_shape());
ngraph::replace_node(n, reshape); ngraph::replace_node(n, reshape);
replaced = true; replaced = true;
} }
......
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