Publishing 2019 R1 content

[platform/upstream/dldt.git] / inference-engine / thirdparty / mkl-dnn / src / cpu / jit_uni_reorder_utils.cpp

diff --git a/inference-engine/thirdparty/mkl-dnn/src/cpu/jit_uni_reorder_utils.cpp b/inference-engine/thirdparty/mkl-dnn/src/cpu/jit_uni_reorder_utils.cpp
index cb9c1d1..cf193c8 100644 (file)
--- a/inference-engine/thirdparty/mkl-dnn/src/cpu/jit_uni_reorder_utils.cpp
+++ b/inference-engine/thirdparty/mkl-dnn/src/cpu/jit_uni_reorder_utils.cpp
@@ -69,8 +69,11 @@ status_t cvt_mem_desc_to_layout_desc(const memory_desc_t &md_,
     case memory_format::any:
     case hwio_s8s8:
     case hwigo_s8s8:
+    case gOIhw4o4i_s8s8:
+    case gOIhw2i8o4i_s8s8:
     case gOIhw4i16o4i_s8s8:
     case OIhw4i16o4i_s8s8:
+    case Goihw16g_s8s8:
     case wino_fmt:
         return invalid_arguments;
     case OIhw4i16o4i:
@@ -107,6 +110,16 @@ status_t cvt_mem_desc_to_layout_desc(const memory_desc_t &md_,
         if (md.format() == OIhw8o16i2o)
             P(3, bd.padding_dims[3], bd.strides[0][3]);
         return success;
+    case gOIhw2i8o4i:
+        P(0, bd.padding_dims[0], bd.strides[0][0]);
+        P(1, bd.padding_dims[1] / 8, bd.strides[0][1]);
+        P(1, 8, 4);
+        P(2, bd.padding_dims[2] / 8, bd.strides[0][2]);
+        P(2, 2, 8*4);
+        P(2, 4, 1);
+        P(3, bd.padding_dims[3], bd.strides[0][3]);
+        P(4, bd.padding_dims[4], bd.strides[0][4]);
+        return success;
     case gOIhw4i16o4i:
         P(0, bd.padding_dims[0], bd.strides[0][0]);
         P(1, bd.padding_dims[1] / 16, bd.strides[0][1]);