diff options
Diffstat (limited to 'src/armnnTfLiteParser')
| -rw-r--r-- | src/armnnTfLiteParser/TfLiteParser.cpp | 1 | ||||
| -rw-r--r-- | src/armnnTfLiteParser/test/FullyConnected.cpp | 4 |
2 files changed, 3 insertions, 2 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); 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 |