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/FullyConnected.cpp

diff --git a/src/armnnTfLiteParser/test/FullyConnected.cpp b/src/armnnTfLiteParser/test/FullyConnected.cpp
index d1223d5..e7aa908 100644
--- a/src/armnnTfLiteParser/test/FullyConnected.cpp
+++ b/src/armnnTfLiteParser/test/FullyConnected.cpp
@@ -171,4 +171,28 @@
             { (40+10)/2, (400+10)/2 });
 }
 
+struct DynamicFullyConnectedWithBiasMultipleOutputsFixture : FullyConnectedFixture
+{
+    DynamicFullyConnectedWithBiasMultipleOutputsFixture()
+        : FullyConnectedFixture("[ 1, 4, 2, 1 ]",     // inputShape
+                                "[ ]",               // outputShape
+                                "[ 1, 4 ]",           // filterShape
+                                "[ 2, 3, 4, 5 ]",     // filterData
+                                "[ 1 ]",              // biasShape
+                                "[ 10, 0, 0, 0 ]" )   // biasData
+    { }
+};
+
+BOOST_FIXTURE_TEST_CASE(
+    DynamicFullyConnectedWithBiasMultipleOutputs,
+    DynamicFullyConnectedWithBiasMultipleOutputsFixture)
+{
+    RunTest<2,
+            armnn::DataType::QAsymmU8,
+            armnn::DataType::QAsymmU8>(0,
+                                      { { "inputTensor", { 1, 2, 3, 4, 10, 20, 30, 40} } },
+                                      { { "outputTensor", { (40+10)/2, (400+10)/2 } } },
+                                      true);
+}
+
 BOOST_AUTO_TEST_SUITE_END()