diff options
author | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2018-10-26 10:24:14 +0100 |
---|---|---|
committer | nattapat.chaimanowong <nattapat.chaimanowong@arm.com> | 2018-10-26 12:38:34 +0000 |
commit | d8eee59735526ead6b87343c3ed9069e682b6e8c (patch) | |
tree | 44b994c98857b28674fb4911cb20489aaaecc437 /src/armnnTfLiteParser/TfLiteParser.cpp | |
parent | d4dfa684941a21314b70593d01b0fc2167eebad4 (diff) | |
download | armnn-d8eee59735526ead6b87343c3ed9069e682b6e8c.tar.gz |
IVGCVSW-2029 Fix fully connected layer support in TfLite Parser and implement test for TfLite VGG16 quantized
Change-Id: I2061f62f62684b963fa0f090718f1dcffe5c93ce
Diffstat (limited to 'src/armnnTfLiteParser/TfLiteParser.cpp')
-rw-r--r-- | src/armnnTfLiteParser/TfLiteParser.cpp | 1 |
1 files changed, 1 insertions, 0 deletions
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); |