[CPU] Fixed port mismatch in Eltwise fusion graph optimization (#20782)

This commit is contained in:
Gorokhov Dmitriy 2023-11-02 11:33:37 +04:00 committed by Alexander Nesterov
parent 63306a167e
commit 0da45a4482

View File

@ -2081,7 +2081,9 @@ void GraphOptimizer::FuseEltwiseAndSimple(Graph &graph) {
graphEdges.push_back(newEdge);
parent->addEdge(newEdge);
parentNode->inputShapes.push_back(parent->getOutputShapeAtPort(inNum));
if (parentNode->inputShapes.size() < static_cast<size_t>(outNum + 1))
parentNode->inputShapes.resize(outNum + 1);
parentNode->inputShapes[outNum] = parent->getOutputShapeAtPort(inNum);
}
}