commit d109a4d0fb980f2e97c05cd86451eda0b53f37ef
author Sadik Armagan <sadik.armagan@arm.com> Tue Jul 28 10:42:13 2020 +0100
committer Sadik Armagan <sadik.armagan@arm.com> Tue Jul 28 16:45:05 2020 +0000
tree 19092204016ccd60ea36637cc06a48a224508570
parent 2f746b3f346d3efa9071fc53592652425869d6b3

IVGCVSW-4932 Introduce ShapeInferenceMethod to TfLite Parser

* Introduced ShapeInferenceMethod::InferAndValidate to TfLiteParser
* Added unit tests

Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: Iad4aadce92912e7645b1652019ec4af478b7fc32

src/armnnTfLiteParser/test/Reshape.cpp

diff --git a/src/armnnTfLiteParser/test/Reshape.cpp b/src/armnnTfLiteParser/test/Reshape.cpp
index 6ed568c..025612f 100644
--- a/src/armnnTfLiteParser/test/Reshape.cpp
+++ b/src/armnnTfLiteParser/test/Reshape.cpp
@@ -138,4 +138,22 @@
                 == armnn::TensorShape({2,3,3})));
 }
 
+struct DynamicReshapeFixtureWithReshapeDimsFlattenOneDim : ReshapeFixture
+{
+    DynamicReshapeFixtureWithReshapeDimsFlattenOneDim() : ReshapeFixture("[ 2, 9 ]",
+                                                                         "[ ]",
+                                                                         "[ 2, -1, 3 ]") {}
+};
+
+BOOST_FIXTURE_TEST_CASE(DynParseReshapeWithReshapeDimsFlattenOneDim, DynamicReshapeFixtureWithReshapeDimsFlattenOneDim)
+{
+    SetupSingleInputSingleOutput("inputTensor", "outputTensor");
+     RunTest<3,
+        armnn::DataType::QAsymmU8,
+        armnn::DataType::QAsymmU8>(0,
+                                   { { "inputTensor", {  1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4, 5, 5, 5, 6, 6, 6 } } },
+                                   { { "outputTensor", { 1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4, 5, 5, 5, 6, 6, 6 } } },
+                                   true);
+}
+
 BOOST_AUTO_TEST_SUITE_END()