diff options
author | Mike Kelly <mike.kelly@arm.com> | 2023-01-19 18:29:40 +0000 |
---|---|---|
committer | mike.kelly <mike.kelly@arm.com> | 2023-01-24 22:57:16 +0000 |
commit | 04d8229bb3e78d1b1dd21eed41e47aabc25d8e2f (patch) | |
tree | 478dbaf477eaa59fac838e6e73b56843f80b52d0 /src/armnnTfLiteParser/test/FullyConnected.cpp | |
parent | 0e3fe10bfe1b4f006f6e0c5c2fae8fb5515c7544 (diff) | |
download | armnn-04d8229bb3e78d1b1dd21eed41e47aabc25d8e2f.tar.gz |
IVGCVSW-7277 Fixed issues with FullyConnected on certain TFLite models
* TFLite Parser:
* Fixed issue in ParseReshape where the targetShape wasn't always calculated correctly
* Fixed issue in ParseFullyConnected where the wrong name was used for the ReshapeLayer
* Added an ExpandDims to the FullyConnected to ensure that we reshape the output correctly
* TFLite Delegate:
* Added an ExpandDims to the FullyConnected to ensure that we reshape the output correctly
Signed-off-by: Mike Kelly <mike.kelly@arm.com>
Change-Id: I129dfcb8543f8a3a297c0589c841be20ef3b6407
Diffstat (limited to 'src/armnnTfLiteParser/test/FullyConnected.cpp')
-rw-r--r-- | src/armnnTfLiteParser/test/FullyConnected.cpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/armnnTfLiteParser/test/FullyConnected.cpp b/src/armnnTfLiteParser/test/FullyConnected.cpp index 108b878e20..05977762c9 100644 --- a/src/armnnTfLiteParser/test/FullyConnected.cpp +++ b/src/armnnTfLiteParser/test/FullyConnected.cpp @@ -1,5 +1,5 @@ // -// Copyright © 2017 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2017-2023 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // @@ -360,7 +360,7 @@ struct FullyConnectedWeightsBiasFloat : FullyConnectedFixture { FullyConnectedWeightsBiasFloat() : FullyConnectedFixture("[ 1, 4, 1, 1 ]", // inputShape - "[ 1, 1 ]", // outputShape + "[ 1, 1, 1, 1 ]", // outputShape "[ 1, 4 ]", // filterShape "[ 2, 3, 4, 5 ]", // filterData "[ 1 ]", // biasShape @@ -373,7 +373,7 @@ struct FullyConnectedWeightsBiasFloat : FullyConnectedFixture TEST_CASE_FIXTURE(FullyConnectedWeightsBiasFloat, "FullyConnectedWeightsBiasFloat") { - RunTest<2, armnn::DataType::Float32>( + RunTest<4, armnn::DataType::Float32>( 0, { 10, 20, 30, 40 }, { 400 }); |