Commit e9e3af0a authored by Liubov Batanina's avatar Liubov Batanina

Add global pool by axis test

parent a33d5008
...@@ -114,10 +114,9 @@ public: ...@@ -114,10 +114,9 @@ public:
virtual bool supportBackend(int backendId) CV_OVERRIDE virtual bool supportBackend(int backendId) CV_OVERRIDE
{ {
return backendId == DNN_BACKEND_OPENCV || return backendId == DNN_BACKEND_OPENCV ||
((backendId == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 || backendId == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH) && (backendId == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && sliceRanges.size() == 1) ||
#ifdef HAVE_INF_ENGINE (backendId == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 &&
INF_ENGINE_VER_MAJOR_GE(INF_ENGINE_RELEASE_2019R1) &&
#endif
sliceRanges.size() == 1 && sliceRanges[0].size() == 4); sliceRanges.size() == 1 && sliceRanges[0].size() == 4);
} }
......
...@@ -121,6 +121,13 @@ public: ...@@ -121,6 +121,13 @@ public:
} }
}; };
TEST_P(Test_TensorFlow_layers, reduce_mean)
{
if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019)
applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER);
runTensorFlowNet("global_pool_by_axis");
}
TEST_P(Test_TensorFlow_layers, conv) TEST_P(Test_TensorFlow_layers, conv)
{ {
runTensorFlowNet("single_conv"); runTensorFlowNet("single_conv");
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment