[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 GitHub
parent 2afb64b9ea
commit 3ff85ae70a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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);
}
}