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/test/FullyConnected.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/test/FullyConnected.cpp')
-rw-r--r-- | src/armnnTfLiteParser/test/FullyConnected.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
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 |