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
// Null broadcast operation, replace with reshape
AxisVector axis_order = ngraph::get_default_order(input_shape);
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);
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