[CPU] AdaptivePooling child edges number check fix (openvinotoolkit#1…

…0372)
mvafin · Feb 15, 2022 · dc905f9 · dc905f9
1 parent fa6865d
commit dc905f9
Showing 1 changed file with 2 additions and 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
@@ -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)) {