Publishing 2019 R1 content

[platform/upstream/dldt.git] / inference-engine / thirdparty / mkl-dnn / src / common / mkldnn_traits.hpp

diff --git a/inference-engine/thirdparty/mkl-dnn/src/common/mkldnn_traits.hpp b/inference-engine/thirdparty/mkl-dnn/src/common/mkldnn_traits.hpp
index f5512b8..367a02a 100644 (file)
--- a/inference-engine/thirdparty/mkl-dnn/src/common/mkldnn_traits.hpp
+++ b/inference-engine/thirdparty/mkl-dnn/src/common/mkldnn_traits.hpp
@@ -39,6 +39,7 @@ template <> struct prec_traits<data_type::s32> { typedef int32_t type; };
 template <> struct prec_traits<data_type::s16> { typedef int16_t type; };
 template <> struct prec_traits<data_type::s8> { typedef int8_t type; };
 template <> struct prec_traits<data_type::u8> { typedef uint8_t type; };
+template <> struct prec_traits<data_type::bin> { typedef uint8_t type; };
 
 template <> struct data_traits<float>
 { static constexpr data_type_t data_type = data_type::f32; };
@@ -71,9 +72,10 @@ PKIND_TRAITS_INST(pooling);
 PKIND_TRAITS_INST(lrn);
 PKIND_TRAITS_INST(batch_normalization);
 PKIND_TRAITS_INST(inner_product);
-PKIND_TRAITS_INST(convolution_relu);
 PKIND_TRAITS_INST(rnn);
 PKIND_TRAITS_INST(roi_pooling);
+PKIND_TRAITS_INST(binary_convolution);
+PKIND_TRAITS_INST(binarization);
 #undef PKIND_TRAITS_INST
 
 }