From 34ba32ff27e7bc967969fe07dcf718a8f8519252 Mon Sep 17 00:00:00 2001 From: Dmitry Kurtaev Date: Sun, 18 Feb 2018 12:45:43 +0300 Subject: [PATCH] Fix DNN flatten test --- modules/dnn/test/test_tf_importer.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/dnn/test/test_tf_importer.cpp b/modules/dnn/test/test_tf_importer.cpp index e6504f1..a2b6df6 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) -- 2.7.4