Publishing 2019 R1 content
[platform/upstream/dldt.git] / inference-engine / thirdparty / mkl-dnn / tests / gtests / in / convolution_simple_small.h
index c9bf46a..f901e9b 100644 (file)
@@ -119,7 +119,7 @@ INST_TEST_CASE(SimpleSmall_Blocked_1x1,
     PARAMS(FMT_DATA_BLOCKED, FMT_WEIGHTS_BLOCKED_G, FMT_BIAS, FMT_DATA_BLOCKED,
         2, 4, 16, 10, 10, 32, 10, 10, 1, 1, 0, 0, 1, 1),
     PARAMS(FMT_DATA_BLOCKED, FMT_WEIGHTS_BLOCKED_G, FMT_BIAS, FMT_DATA_BLOCKED,
-        2, 8, 32, 10, 10, 256, 10, 10, 1, 1, 0, 0, 1, 1)
+        1, 8, 32, 1, 1, 128, 1, 1, 1, 1, 0, 0, 1, 1)
 );
 
 INST_TEST_CASE(SimpleSmall_Blocked16,
@@ -164,6 +164,12 @@ INST_TEST_CASE(SimpleSmall_Blocked16,
 );
 
 INST_TEST_CASE(SimpleSmall_Regression,
+    /* grouped small input-channel avx512 */
+    PARAMS(nchw, gOhwi16o, FMT_BIAS, FMT_DATA_BLOCKED16,
+        2, 2, 16, 8, 8, 32, 8, 8, 3, 3, 1, 1, 1, 1),
+    /* grouped small input-channel avx2 */
+    PARAMS(nchw, gOhwi8o, FMT_BIAS, nChw8c,
+        2, 2, 4, 2, 2, 16, 8, 8, 3, 3, 1, 1, 1, 1),
     PARAMS(FMT_DATA_BLOCKED16, FMT_WEIGHTS_BLOCKED16, FMT_BIAS, FMT_DATA_BLOCKED16,
         2, 1, 32, 16, 16, 32, 16, 16, 3, 3, 0, 0, 1, 1),
     PARAMS(FMT_DATA_BLOCKED16, FMT_WEIGHTS_BLOCKED16, FMT_BIAS, FMT_DATA_BLOCKED16,