Fix DNN flatten test

7 years ago · 34ba32ff27
parent 857007372d
commit 34ba32ff27
1 changed files with 1 additions and 1 deletions
--- 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)