diff --git a/modules/dnn/test/test_tf_importer.cpp b/modules/dnn/test/test_tf_importer.cpp index e6504f179e..a2b6df6e93 100644 --- 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)