Fix DNN flatten test

dkurt · dkurt · commit 34ba32ff27e7 · 2018-02-18T12:45:43.000+03:00
diff --git 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)

Original file line number	Diff line number	Diff line change
`@@ -199,7 +199,7 @@ TEST(Test_TensorFlow, reshape)`
`199`	`199`	`{`
`200`	`200`	`runTensorFlowNet("shift_reshape_no_reorder");`
`201`	`201`	`runTensorFlowNet("reshape_reduce");`
`202`		`- runTensorFlowNet("flatten", true);`
	`202`	`+ runTensorFlowNet("flatten", DNN_TARGET_CPU, true);`
`203`	`203`	`}`
`204`	`204`
`205`	`205`	`TEST(Test_TensorFlow, fp16)`