From 045d26ff31998cbe47d3213be6f863c905fee60f Mon Sep 17 00:00:00 2001
From: Alexander Peskov <alexander.peskov@intel.com>
Date: Thu, 29 Oct 2020 00:15:53 +0300
Subject: [PATCH] [CPU] Fix case with unused port in Loop body

Signed-off-by: Alexander Peskov <alexander.peskov@intel.com>
---
 inference-engine/src/mkldnn_plugin/mkldnn_graph.cpp | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/inference-engine/src/mkldnn_plugin/mkldnn_graph.cpp b/inference-engine/src/mkldnn_plugin/mkldnn_graph.cpp
index f072a3426d2..d6d1f483c85 100644
--- a/inference-engine/src/mkldnn_plugin/mkldnn_graph.cpp
+++ b/inference-engine/src/mkldnn_plugin/mkldnn_graph.cpp
@@ -132,6 +132,10 @@ void MKLDNNGraph::Replicate(const TensorIterator::Body &subgraph, const MKLDNNEx
 
         graphNodes.push_back(node);
         inputNodes[input->getName()] = node;
+
+        if (getInputTo(input).empty()) {
+            unused_data.insert(input);
+        }
     }
 
     // Step 2. Replicate all internal nodes.