Fix DNN flatten test

34ba32ff · Dmitry Kurtaev · 85700737 · 34ba32ff
Commit 34ba32ff authored Feb 18, 2018 by Dmitry Kurtaev
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

test_tf_importer.cpp modules/dnn/test/test_tf_importer.cpp +1 -1

No files found.
--- a/modules/dnn/test/test_tf_importer.cpp
+++ b/modules/dnn/test/test_tf_importer.cpp
@@ -199,7 +199,7 @@ TEST(Test_TensorFlow, reshape)
 {
    runTensorFlowNet("shift_reshape_no_reorder");
    runTensorFlowNet("reshape_reduce");
-    runTensorFlowNet("flatten", true);
+    runTensorFlowNet("flatten", DNN_TARGET_CPU, true);
 }
 TEST(Test_TensorFlow, fp16)