Rebase onto ngraph migration

openvinotoolkit · May 7, 2021 · aed9f51 · aed9f51
1 parent cc743e7
commit aed9f51
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 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
@@ -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
@@ -4,8 +4,9 @@
 
 #pragma once
 
-#include <ie_common.h>
 #include <mkldnn_node.h>
+#include <ie_common.h>
+
 #include <string>
 #include <memory>
 #include <vector>