diff options
| author | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-01-25 13:20:39 +0000 |
|---|---|---|
| committer | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-01-25 13:20:39 +0000 |
| commit | 5e9d29802e2cfbb13adc49c2a0ac9ba952dc7650 (patch) | |
| tree | 31baaf01ff767159fb6a4e594405b4acf03a6f51 /src/armnnTfLiteParser/test/Concatenation.cpp | |
| parent | 6e2f60674cbe77c2a1da94ab71e35c298a1924de (diff) | |
| download | armnn-5e9d29802e2cfbb13adc49c2a0ac9ba952dc7650.tar.gz | |
IVGCVSW-2563 Fix bug in TfLiteParser::ParseConcatenation
Change-Id: I8fbf27b383a821e062f72809cc2e269fcd18851c
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() |