diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-04-24 15:52:20 +0100 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-04-29 08:36:55 +0000 |
commit | 501f4d4efff78f890602d062709126f9a294a352 (patch) | |
tree | 259dcc6bb6fcd895841c276ebc5025c756f656e3 /src/armnnTfLiteParser/test | |
parent | c01b39146d4b67159ef95a2ac2f88f6536320890 (diff) | |
download | armnn-501f4d4efff78f890602d062709126f9a294a352.tar.gz |
IVGCVSW-2996 Add Reshape layer to ParseFullyConnected in TfLite parser
when input is > 2D to flatten the input to 2D [batch_size, input_size]
Change-Id: Id9d9ff996225c7d0938204ae0ceb330a11e264f5
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Diffstat (limited to 'src/armnnTfLiteParser/test')
-rw-r--r-- | src/armnnTfLiteParser/test/FullyConnected.cpp | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/src/armnnTfLiteParser/test/FullyConnected.cpp b/src/armnnTfLiteParser/test/FullyConnected.cpp index 7ee64a476e..54d7bcb1dc 100644 --- a/src/armnnTfLiteParser/test/FullyConnected.cpp +++ b/src/armnnTfLiteParser/test/FullyConnected.cpp @@ -151,4 +151,24 @@ BOOST_FIXTURE_TEST_CASE(ParseFullyConnectedWithBias, FullyConnectedWithBiasFixtu { (400+10)/2 }); } +struct FullyConnectedWithBiasMultipleOutputsFixture : FullyConnectedFixture +{ + FullyConnectedWithBiasMultipleOutputsFixture() + : FullyConnectedFixture("[ 1, 4, 2, 1 ]", // inputShape + "[ 2, 1 ]", // outputShape + "[ 1, 4 ]", // filterShape + "[ 2, 3, 4, 5 ]", // filterData + "[ 1 ]", // biasShape + "[ 10, 0, 0, 0 ]" ) // biasData + {} +}; + +BOOST_FIXTURE_TEST_CASE(FullyConnectedWithBiasMultipleOutputs, FullyConnectedWithBiasMultipleOutputsFixture) +{ + RunTest<2, armnn::DataType::QuantisedAsymm8>( + 0, + { 1, 2, 3, 4, 10, 20, 30, 40 }, + { (40+10)/2, (400+10)/2 }); +} + BOOST_AUTO_TEST_SUITE_END() |