From 4be0c595059e0eab6166b174f3c9514cbb945d83 Mon Sep 17 00:00:00 2001 From: Luwei Zhou Date: Mon, 27 Jun 2022 10:43:38 +0800 Subject: [PATCH] Fix the Non_Zero childedge check. (#11963) --- src/plugins/intel_cpu/src/nodes/non_zero.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/plugins/intel_cpu/src/nodes/non_zero.cpp b/src/plugins/intel_cpu/src/nodes/non_zero.cpp index 76d6734f79f..0e7063e80b4 100644 --- a/src/plugins/intel_cpu/src/nodes/non_zero.cpp +++ b/src/plugins/intel_cpu/src/nodes/non_zero.cpp @@ -42,7 +42,7 @@ void NonZero::getSupportedDescriptors() { return; if (getParentEdges().size() != 1) IE_THROW() << errorPrefix << "has incorrect number of input edges: " << getParentEdges().size(); - if (getChildEdges().size() != 1) + if (!getChildEdges().size()) IE_THROW() << errorPrefix << "has incorrect number of output edges: " << getChildEdges().size(); }