Apply review comments 2

openvinotoolkit · May 24, 2021 · 628c55f · 628c55f
1 parent fa795ac
commit 628c55f
Showing 1 changed file with 2 additions and 2 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
@@ -448,8 +448,8 @@ void MKLDNNROIPoolingNode::createPrimitive() {
     auto selectedPD = getSelectedPrimitiveDescriptor();
     jpp.src_prc = selectedPD->getConfig().inConfs[0].desc.getPrecision();
     jpp.dst_prc = selectedPD->getConfig().outConfs[0].desc.getPrecision();
-    jpp.src_data_size = MKLDNNExtensionUtils::sizeOfDataType(MKLDNNExtensionUtils::IEPrecisionToDataType(jpp.src_prc));
-    jpp.dst_data_size = MKLDNNExtensionUtils::sizeOfDataType(MKLDNNExtensionUtils::IEPrecisionToDataType(jpp.dst_prc));
+    jpp.src_data_size = jpp.src_prc.size();
+    jpp.dst_data_size = jpp.dst_prc.size();
 
     jpp.alg = getAlgorithm();