Publishing 2019 R1 content

[platform/upstream/dldt.git] / inference-engine / src / mkldnn_plugin / mkldnn_extension_utils.cpp

diff --git a/inference-engine/src/mkldnn_plugin/mkldnn_extension_utils.cpp b/inference-engine/src/mkldnn_plugin/mkldnn_extension_utils.cpp
index 3600ee5..7b45731 100644 (file)
--- a/inference-engine/src/mkldnn_plugin/mkldnn_extension_utils.cpp
+++ b/inference-engine/src/mkldnn_plugin/mkldnn_extension_utils.cpp
@@ -1,4 +1,4 @@
-// Copyright (C) 2018 Intel Corporation
+// Copyright (C) 2018-2019 Intel Corporation
 // SPDX-License-Identifier: Apache-2.0
 //
 
@@ -21,8 +21,11 @@ uint8_t MKLDNNExtensionUtils::sizeOfDataType(mkldnn::memory::data_type dataType)
         return 1;
     case mkldnn::memory::data_type::u8:
         return 1;
+    case mkldnn::memory::data_type::bin:
+        return 1;
     case mkldnn::memory::data_type::data_undef:
         return 0;
+
     default:
         THROW_IE_EXCEPTION << "Unsupported data type.";
     }
@@ -40,6 +43,8 @@ memory::data_type MKLDNNExtensionUtils::IEPrecisionToDataType(InferenceEngine::P
             return memory::s8;
         case InferenceEngine::Precision::U8:
             return memory::u8;
+        case InferenceEngine::Precision::BIN:
+            return memory::bin;
 
         default: {
             THROW_IE_EXCEPTION << "The plugin does not support " << prec.name();
@@ -59,6 +64,8 @@ InferenceEngine::Precision MKLDNNExtensionUtils::DataTypeToIEPrecision(memory::d
             return InferenceEngine::Precision::I8;
         case memory::u8:
             return InferenceEngine::Precision::U8;
+        case memory::bin:
+            return InferenceEngine::Precision::BIN;
 
         default: {
             THROW_IE_EXCEPTION << "Unsupported data type.";