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/TfLiteParser.cpp | 1 + src/armnnTfLiteParser/test/FullyConnected.cpp | 4 ++-- 2 files changed, 3 insertions(+), 2 deletions(-) (limited to 'src') diff --git a/src/armnnTfLiteParser/TfLiteParser.cpp b/src/armnnTfLiteParser/TfLiteParser.cpp index 8b1d3e6bc4..5e0d4b7d6a 100644 --- a/src/armnnTfLiteParser/TfLiteParser.cpp +++ b/src/armnnTfLiteParser/TfLiteParser.cpp @@ -1231,6 +1231,7 @@ void TfLiteParser::ParseFullyConnected(size_t subgraphIndex, size_t operatorInde FullyConnectedDescriptor desc; desc.m_BiasEnabled = false; + desc.m_TransposeWeightMatrix = true; auto inputs = GetInputs(m_Model, subgraphIndex, operatorIndex); auto outputs = GetOutputs(m_Model, subgraphIndex, operatorIndex); 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