Merge pull request #10896 from dkurt:tensorflow_tests_generator

7 years ago · 68043f4de9
parent c0eaff3608 34ba32ff27
commit 68043f4de9
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)