From 4ef700c6e4bd8235802b35bcdb7faaeea152d6a4 Mon Sep 17 00:00:00 2001
From: Anton Voronov <anton.voronov@intel.com>
Date: Thu, 26 Aug 2021 13:38:24 +0300
Subject: [PATCH] [CPU] fixed convolution outputShape in ConvDWConv fusing
 (#7239)

---
 inference-engine/src/mkldnn_plugin/mkldnn_graph.cpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/inference-engine/src/mkldnn_plugin/mkldnn_graph.cpp b/inference-engine/src/mkldnn_plugin/mkldnn_graph.cpp
index 2a41d514b8f..eb6f5fc523b 100644
--- a/inference-engine/src/mkldnn_plugin/mkldnn_graph.cpp
+++ b/inference-engine/src/mkldnn_plugin/mkldnn_graph.cpp
@@ -1028,6 +1028,8 @@ void MKLDNNGraph::DropDWConvNode(const MKLDNNNodePtr &node) {
     auto parentConv = parentConvEdge->getParent();
     if (!parentConv) return;
 
+    parentConv->outputShapes[0] = node->outputShapes[0];
+
     for (size_t i = 0; i < 1; i++) {
         auto p_edge = parents[i].lock();
         if (!p_edge) continue;