summaryrefslogtreecommitdiff
path: root/inference-engine/thirdparty/mkl-dnn/tests/benchdnn/inputs/test_conv_all
diff options
context:
space:
mode:
Diffstat (limited to 'inference-engine/thirdparty/mkl-dnn/tests/benchdnn/inputs/test_conv_all')
-rw-r--r--inference-engine/thirdparty/mkl-dnn/tests/benchdnn/inputs/test_conv_all4
1 files changed, 4 insertions, 0 deletions
diff --git a/inference-engine/thirdparty/mkl-dnn/tests/benchdnn/inputs/test_conv_all b/inference-engine/thirdparty/mkl-dnn/tests/benchdnn/inputs/test_conv_all
index 5af9c4a2d..5354842c6 100644
--- a/inference-engine/thirdparty/mkl-dnn/tests/benchdnn/inputs/test_conv_all
+++ b/inference-engine/thirdparty/mkl-dnn/tests/benchdnn/inputs/test_conv_all
@@ -39,6 +39,7 @@
--match=.*kh3[^0-9].* # only 3x3 convolutions so far
--allow-unimpl=true # allow unimplemented for groups > 1
--mb=2 # for fwd and bwd_d reduce mb
+--dir=FWD_I --batch=conv_all
--dir=FWD_B --batch=conv_all
--dir=BWD_D --batch=conv_all
--dir=BWD_WB --batch=conv_all
@@ -63,3 +64,6 @@
# tails
--batch=test_conv_tails
+
+# 3D conv
+--batch=test_conv_3d \ No newline at end of file