diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-04-23 15:28:06 +0100 |
---|---|---|
committer | Ruomei Yan <ruomei.yan@arm.com> | 2019-04-25 10:26:19 +0000 |
commit | 672de578c819a3d815e037cddfaf4dcf49d12917 (patch) | |
tree | a85e588ce044b4733919b05bff0244d01a19cb93 /src/armnnTfLiteParser/test/Reshape.cpp | |
parent | 4a95611748583f35f0f1f8388a3ba261e63436b7 (diff) | |
download | armnn-672de578c819a3d815e037cddfaf4dcf49d12917.tar.gz |
IVGCVSW-2994 Add Reshape layer to ParseUnpack in TfLite parser
to remove the unpacked dimension of each output from Splitter
and correct ReshapeFixtureWithReshapeDimsFlatten test output shape
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I517d315475612ac8b773930f9b58cac316fa8553
Diffstat (limited to 'src/armnnTfLiteParser/test/Reshape.cpp')
-rw-r--r-- | src/armnnTfLiteParser/test/Reshape.cpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/armnnTfLiteParser/test/Reshape.cpp b/src/armnnTfLiteParser/test/Reshape.cpp index ef4b761945..62fbad6953 100644 --- a/src/armnnTfLiteParser/test/Reshape.cpp +++ b/src/armnnTfLiteParser/test/Reshape.cpp @@ -95,17 +95,17 @@ BOOST_FIXTURE_TEST_CASE(ParseReshapeWithReshapeDims, ReshapeFixtureWithReshapeDi struct ReshapeFixtureWithReshapeDimsFlatten : ReshapeFixture { - ReshapeFixtureWithReshapeDimsFlatten() : ReshapeFixture("[ 3, 3 ]", "[ 1, 9 ]", "[ -1 ]") {} + ReshapeFixtureWithReshapeDimsFlatten() : ReshapeFixture("[ 3, 3 ]", "[ 9 ]", "[ -1 ]") {} }; BOOST_FIXTURE_TEST_CASE(ParseReshapeWithReshapeDimsFlatten, ReshapeFixtureWithReshapeDimsFlatten) { SetupSingleInputSingleOutput("inputTensor", "outputTensor"); - RunTest<2, armnn::DataType::QuantisedAsymm8>(0, + RunTest<1, armnn::DataType::QuantisedAsymm8>(0, { 1, 2, 3, 4, 5, 6, 7, 8, 9 }, { 1, 2, 3, 4, 5, 6, 7, 8, 9 }); BOOST_TEST((m_Parser->GetNetworkOutputBindingInfo(0, "outputTensor").second.GetShape() - == armnn::TensorShape({1,9}))); + == armnn::TensorShape({9}))); } struct ReshapeFixtureWithReshapeDimsFlattenTwoDims : ReshapeFixture |