Fix DNN flatten test

This commit is contained in:
Dmitry Kurtaev 2018-02-18 12:45:43 +03:00
parent 857007372d
commit 34ba32ff27

View File

@ -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)