DeformableConv: Decrease input_shape and kernal in CPU functional tests (#6746)

This commit is contained in:
Ivan Tikhonov 2021-07-22 01:37:22 +03:00 committed by GitHub
parent a3825ba1a5
commit 9f7cf581b8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -22,8 +22,8 @@ const std::vector<size_t> groups = {1};
const std::vector<size_t> defor_groups = {2}; const std::vector<size_t> defor_groups = {2};
const std::vector<size_t> numOutChannels = {1, 5}; const std::vector<size_t> numOutChannels = {1, 5};
const std::vector<size_t> multiple_defor_groups = {4}; const std::vector<size_t> multiple_defor_groups = {4};
const std::vector<std::vector<size_t>> deform_vals = {{1, 200, 220, 220}}; const std::vector<std::vector<size_t>> deform_vals = {{1, 72, 64, 64}};
const std::vector<std::vector<size_t>> kernel = {{64, 16, 5, 5}}; const std::vector<std::vector<size_t>> kernel = {{16, 16, 3, 3}};
const std::vector<bool> with_bilinear_interpolation_pad = { false, true }; const std::vector<bool> with_bilinear_interpolation_pad = { false, true };
const std::vector<bool> with_modulated_scalar = { false, true }; const std::vector<bool> with_modulated_scalar = { false, true };
@ -92,7 +92,7 @@ INSTANTIATE_TEST_SUITE_P(
::testing::Values(InferenceEngine::Precision::UNSPECIFIED), ::testing::Values(InferenceEngine::Precision::UNSPECIFIED),
::testing::Values(InferenceEngine::Layout::ANY), ::testing::Values(InferenceEngine::Layout::ANY),
::testing::Values(InferenceEngine::Layout::ANY), ::testing::Values(InferenceEngine::Layout::ANY),
::testing::Values(std::vector<size_t>({1, 16, 224, 224})), ::testing::Values(std::vector<size_t>({1, 16, 66, 66})),
::testing::Values(CommonTestUtils::DEVICE_CPU)), ::testing::Values(CommonTestUtils::DEVICE_CPU)),
DeformableConvolutionLayerTest::getTestCaseName); DeformableConvolutionLayerTest::getTestCaseName);