X-Git-Url: http://review.tizen.org/git/?a=blobdiff_plain;f=modules%2Fdnn%2Ftest%2Ftest_tf_importer.cpp;h=7757011cffab5de1dacad3ab6fc87fee9eb97b6d;hb=613c12e59015f4bd7909916ceee195edd7ef88d0;hp=ea8cd1565945872346def47d31df9a8072a5b210;hpb=fd11e3a81d6347e8dfe5b04f41dee315327f8005;p=platform%2Fupstream%2Fopencv.git diff --git a/modules/dnn/test/test_tf_importer.cpp b/modules/dnn/test/test_tf_importer.cpp index ea8cd15..7757011 100644 --- a/modules/dnn/test/test_tf_importer.cpp +++ b/modules/dnn/test/test_tf_importer.cpp @@ -136,8 +136,8 @@ TEST_P(Test_TensorFlow_layers, Convolution3D) #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_LT(2019010000) throw SkipTestException("Test is enabled starts from 2019R1"); #endif - if (target != DNN_TARGET_CPU) - throw SkipTestException("Only CPU is supported"); + if (target != DNN_TARGET_CPU && backend != DNN_BACKEND_CUDA) + throw SkipTestException("Only CPU and CUDA is supported"); runTensorFlowNet("conv3d"); } @@ -252,8 +252,8 @@ TEST_P(Test_TensorFlow_layers, MaxPooling3D) #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_LT(2019010000) throw SkipTestException("Test is enabled starts from 2019R1"); #endif - if (target != DNN_TARGET_CPU) - throw SkipTestException("Only CPU is supported"); + if (target != DNN_TARGET_CPU && backend != DNN_BACKEND_CUDA) + throw SkipTestException("Only CPU and CUDA is supported"); runTensorFlowNet("max_pool3d"); } @@ -262,13 +262,16 @@ TEST_P(Test_TensorFlow_layers, AvePooling3D) #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_LT(2019010000) throw SkipTestException("Test is enabled starts from 2019R1"); #endif - if (target != DNN_TARGET_CPU) - throw SkipTestException("Only CPU is supported"); + if (target != DNN_TARGET_CPU && backend != DNN_BACKEND_CUDA) + throw SkipTestException("Only CPU and CUDA is supported"); runTensorFlowNet("ave_pool3d"); } TEST_P(Test_TensorFlow_layers, deconvolution) { + if(backend == DNN_BACKEND_CUDA) + applyTestTag(CV_TEST_TAG_DNN_SKIP_CUDA); /* bugged */ + runTensorFlowNet("deconvolution"); runTensorFlowNet("deconvolution_same"); runTensorFlowNet("deconvolution_stride_2_same"); @@ -679,6 +682,8 @@ TEST_P(Test_TensorFlow_layers, quantized) TEST_P(Test_TensorFlow_layers, lstm) { + if(backend == DNN_BACKEND_CUDA) + applyTestTag(CV_TEST_TAG_DNN_SKIP_CUDA); /* not supported */ if (backend == DNN_BACKEND_INFERENCE_ENGINE) applyTestTag(CV_TEST_TAG_DNN_SKIP_IE); if (backend == DNN_BACKEND_OPENCV && target == DNN_TARGET_OPENCL_FP16)