From d8eee59735526ead6b87343c3ed9069e682b6e8c Mon Sep 17 00:00:00 2001 From: Nattapat Chaimanowong Date: Fri, 26 Oct 2018 10:24:14 +0100 Subject: 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 | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src/armnnTfLiteParser/test') diff --git a/src/armnnTfLiteParser/test/FullyConnected.cpp b/src/armnnTfLiteParser/test/FullyConnected.cpp index 2853fe96ab..14ca57c2ab 100644 --- a/src/armnnTfLiteParser/test/FullyConnected.cpp +++ b/src/armnnTfLiteParser/test/FullyConnected.cpp @@ -118,7 +118,7 @@ struct FullyConnectedWithNoBiasFixture : FullyConnectedFixture 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 @@ struct FullyConnectedWithBiasFixture : FullyConnectedFixture 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 -- cgit v1.2.1