diff options
Diffstat (limited to 'src/armnnTfLiteParser/test/Concatenation.cpp')
-rw-r--r-- | src/armnnTfLiteParser/test/Concatenation.cpp | 51 |
1 files changed, 51 insertions, 0 deletions
diff --git a/src/armnnTfLiteParser/test/Concatenation.cpp b/src/armnnTfLiteParser/test/Concatenation.cpp index bb5aebf39c..d3d571f174 100644 --- a/src/armnnTfLiteParser/test/Concatenation.cpp +++ b/src/armnnTfLiteParser/test/Concatenation.cpp @@ -189,4 +189,55 @@ BOOST_FIXTURE_TEST_CASE(ParseConcatenationDim3, ConcatenationFixtureDim3) 70, 71, 72, 73 } } }); } +struct ConcatenationFixture3DDim0 : ConcatenationFixture +{ + ConcatenationFixture3DDim0() : ConcatenationFixture("[ 1, 2, 3]", "[ 2, 2, 3]", "[ 3, 2, 3]", "0" ) {} +}; + +BOOST_FIXTURE_TEST_CASE(ParseConcatenation3DDim0, ConcatenationFixture3DDim0) +{ + RunTest<3, armnn::DataType::QuantisedAsymm8>( + 0, + { { "inputTensor1", { 0, 1, 2, 3, 4, 5 } }, + { "inputTensor2", { 6, 7, 8, 9, 10, 11, + 12, 13, 14, 15, 16, 17 } } }, + { { "outputTensor", { 0, 1, 2, 3, 4, 5, + 6, 7, 8, 9, 10, 11, + 12, 13, 14, 15, 16, 17 } } }); +} + +struct ConcatenationFixture3DDim1 : ConcatenationFixture +{ + ConcatenationFixture3DDim1() : ConcatenationFixture("[ 1, 2, 3]", "[ 1, 4, 3]", "[ 1, 6, 3]", "1" ) {} +}; + +BOOST_FIXTURE_TEST_CASE(ParseConcatenation3DDim1, ConcatenationFixture3DDim1) +{ + RunTest<3, armnn::DataType::QuantisedAsymm8>( + 0, + { { "inputTensor1", { 0, 1, 2, 3, 4, 5 } }, + { "inputTensor2", { 6, 7, 8, 9, 10, 11, + 12, 13, 14, 15, 16, 17 } } }, + { { "outputTensor", { 0, 1, 2, 3, 4, 5, + 6, 7, 8, 9, 10, 11, + 12, 13, 14, 15, 16, 17 } } }); +} + +struct ConcatenationFixture3DDim2 : ConcatenationFixture +{ + ConcatenationFixture3DDim2() : ConcatenationFixture("[ 1, 2, 3]", "[ 1, 2, 6]", "[ 1, 2, 9]", "2" ) {} +}; + +BOOST_FIXTURE_TEST_CASE(ParseConcatenation3DDim2, ConcatenationFixture3DDim2) +{ + RunTest<3, armnn::DataType::QuantisedAsymm8>( + 0, + { { "inputTensor1", { 0, 1, 2, + 3, 4, 5 } }, + { "inputTensor2", { 6, 7, 8, 9, 10, 11, + 12, 13, 14, 15, 16, 17 } } }, + { { "outputTensor", { 0, 1, 2, 6, 7, 8, 9, 10, 11, + 3, 4, 5, 12, 13, 14, 15, 16, 17 } } }); +} + BOOST_AUTO_TEST_SUITE_END() |