commit d8eee59735526ead6b87343c3ed9069e682b6e8c
author Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> Fri Oct 26 10:24:14 2018 +0100
committer nattapat.chaimanowong <nattapat.chaimanowong@arm.com> Fri Oct 26 12:38:34 2018 +0000
tree 44b994c98857b28674fb4911cb20489aaaecc437
parent d4dfa684941a21314b70593d01b0fc2167eebad4

IVGCVSW-2029 Fix fully connected layer support in TfLite Parser and implement test for TfLite VGG16 quantized

Change-Id: I2061f62f62684b963fa0f090718f1dcffe5c93ce

src/armnnTfLiteParser/test/FullyConnected.cpp

diff --git a/src/armnnTfLiteParser/test/FullyConnected.cpp b/src/armnnTfLiteParser/test/FullyConnected.cpp
index 2853fe9..14ca57c 100644
--- a/src/armnnTfLiteParser/test/FullyConnected.cpp
+++ b/src/armnnTfLiteParser/test/FullyConnected.cpp
@@ -118,7 +118,7 @@
     FullyConnectedWithNoBiasFixture()
         : FullyConnectedFixture("[ 1, 4, 1, 1 ]",     // inputShape
                                 "[ 1, 1 ]",           // outputShape
-                                "[ 4, 1 ]",           // filterShape
+                                "[ 1, 4 ]",           // filterShape
                                 "[ 2, 3, 4, 5 ]")     // filterData
     {}
 };
@@ -136,7 +136,7 @@
     FullyConnectedWithBiasFixture()
         : FullyConnectedFixture("[ 1, 4, 1, 1 ]",     // inputShape
                                 "[ 1, 1 ]",           // outputShape
-                                "[ 4, 1 ]",           // filterShape
+                                "[ 1, 4 ]",           // filterShape
                                 "[ 2, 3, 4, 5 ]",     // filterData
                                 "[ 1 ]",              // biasShape
                                 "[ 10, 0, 0, 0 ]" )   // biasData