Skip to content

Commit

Permalink
[IE CLDNN] Fixed CTCGreedyDecoderSeqLenLayerTest gpu instances (#4326)
Browse files Browse the repository at this point in the history
  • Loading branch information
vladimir-paramuzov authored Feb 14, 2021
1 parent 434e66d commit 51a7d7b
Showing 1 changed file with 3 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -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),
Expand All @@ -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}),
Expand Down

0 comments on commit 51a7d7b

Please sign in to comment.