From e6e9c20ee94c184de2a6e924ea57998ea3024244 Mon Sep 17 00:00:00 2001 From: mandrono Date: Tue, 15 Feb 2022 11:07:54 +0300 Subject: [PATCH] [CPU] AdaptivePooling child edges number check fix --- src/plugins/intel_cpu/src/nodes/mkldnn_adaptive_pooling.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/plugins/intel_cpu/src/nodes/mkldnn_adaptive_pooling.cpp b/src/plugins/intel_cpu/src/nodes/mkldnn_adaptive_pooling.cpp index d1821ce97de3b7..817b380a6e99c5 100644 --- a/src/plugins/intel_cpu/src/nodes/mkldnn_adaptive_pooling.cpp +++ b/src/plugins/intel_cpu/src/nodes/mkldnn_adaptive_pooling.cpp @@ -68,8 +68,8 @@ void MKLDNNAdaptivePoolingNode::getSupportedDescriptors() { if (getParentEdges().size() != 2) IE_THROW() << errorPrefix << "has incorrect number of input edges: " << getParentEdges().size(); - if (getChildEdges().size() != (algorithm == AdaptivePoolingMax ? 2 : 1)) - IE_THROW() << errorPrefix << "has incorrect number of output edges: " << getParentEdges().size(); + if (getChildEdges().size() < (algorithm == AdaptivePoolingMax ? 2 : 1)) + IE_THROW() << errorPrefix << "has incorrect number of output edges: " << getChildEdges().size(); auto srcRank = getInputShapeAtPort(0).getRank(); if (!one_of(spatialDimsCount, 1, 2, 3)) {