From aed9f519f2a69354a4bfa1dad3d3c65c500b27e2 Mon Sep 17 00:00:00 2001
From: Egor Duplensky <egor.duplenskii@intel.com>
Date: Fri, 7 May 2021 21:04:52 +0300
Subject: [PATCH] Rebase onto ngraph migration

---
 .../src/mkldnn_plugin/nodes/mkldnn_roi_pooling_node.cpp    | 7 ++++---
 .../src/mkldnn_plugin/nodes/mkldnn_roi_pooling_node.h      | 3 ++-
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/inference-engine/src/mkldnn_plugin/nodes/mkldnn_roi_pooling_node.cpp b/inference-engine/src/mkldnn_plugin/nodes/mkldnn_roi_pooling_node.cpp
index 8dbd350d253b9a..7c8dae85c448aa 100644
--- a/inference-engine/src/mkldnn_plugin/nodes/mkldnn_roi_pooling_node.cpp
+++ b/inference-engine/src/mkldnn_plugin/nodes/mkldnn_roi_pooling_node.cpp
@@ -10,7 +10,6 @@
 
 #include <ngraph/opsets/opset2.hpp>
 
-#include <legacy/ie_layers.h>
 #include "ie_parallel.hpp"
 #include "utils/bfloat16.hpp"
 #include "emitters/jit_load_store_emitters.hpp"
@@ -19,7 +18,9 @@
 
 #include <string>
 #include <vector>
-#include <math.h>
+#include <memory>
+#include <algorithm>
+#include <cmath>
 
 using namespace MKLDNNPlugin;
 using namespace InferenceEngine;
@@ -375,7 +376,7 @@ void MKLDNNROIPoolingNode::initSupportedPrimitiveDescriptors() {
     if (!supportedPrimitiveDescriptors.empty())
         return;
 
-    runtimePrecision = getCnnLayer()->insData[0].lock()->getPrecision();
+    runtimePrecision = getOriginalInputPrecisionAtPort(0);
 
     if (!mayiuse(avx512_core)) {
         if (runtimePrecision == Precision::BF16)
diff --git a/inference-engine/src/mkldnn_plugin/nodes/mkldnn_roi_pooling_node.h b/inference-engine/src/mkldnn_plugin/nodes/mkldnn_roi_pooling_node.h
index 6a04dd4c2c2b79..7028e33e83de29 100644
--- a/inference-engine/src/mkldnn_plugin/nodes/mkldnn_roi_pooling_node.h
+++ b/inference-engine/src/mkldnn_plugin/nodes/mkldnn_roi_pooling_node.h
@@ -4,8 +4,9 @@
 
 #pragma once
 
-#include <ie_common.h>
 #include <mkldnn_node.h>
+#include <ie_common.h>
+
 #include <string>
 #include <memory>
 #include <vector>