[IE CLDNN] Fixed CTCGreedyDecoderSeqLenLayerTest gpu instances (#4326)

This commit is contained in:
Vladimir Paramuzov 2021-02-14 21:30:04 +03:00 committed by GitHub
parent 434e66d7fb
commit 51a7d7b799
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -26,6 +26,7 @@ std::vector<bool> mergeRepeated{true, false};
const auto basicCases = ::testing::Combine(
::testing::ValuesIn(inputShape),
::testing::Values(10),
::testing::ValuesIn(probPrecisions),
::testing::ValuesIn(idxPrecisions),
::testing::Values(0),
@ -38,7 +39,8 @@ INSTANTIATE_TEST_CASE_P(smoke_set1, CTCGreedyDecoderSeqLenLayerTest,
INSTANTIATE_TEST_CASE_P(smoke_set2, CTCGreedyDecoderSeqLenLayerTest,
::testing::Combine(
::testing::Values(std::vector<size_t>{2, 8, 11}),
::testing::ValuesIn(std::vector<std::vector<size_t>>{{2, 8, 11}, {4, 10, 55}}),
::testing::ValuesIn(std::vector<int>{5, 100}),
::testing::ValuesIn(probPrecisions),
::testing::ValuesIn(idxPrecisions),
::testing::ValuesIn(std::vector<int>{0, 5, 10}),