From 5e9d29802e2cfbb13adc49c2a0ac9ba952dc7650 Mon Sep 17 00:00:00 2001 From: Nattapat Chaimanowong Date: Fri, 25 Jan 2019 13:20:39 +0000 Subject: IVGCVSW-2563 Fix bug in TfLiteParser::ParseConcatenation Change-Id: I8fbf27b383a821e062f72809cc2e269fcd18851c --- src/armnnTfLiteParser/test/Concatenation.cpp | 51 ++++++++++++++++++++++++++++ 1 file changed, 51 insertions(+) (limited to 'src/armnnTfLiteParser/test') 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() -- cgit v1.2.1