From 04d8229bb3e78d1b1dd21eed41e47aabc25d8e2f Mon Sep 17 00:00:00 2001 From: Mike Kelly Date: Thu, 19 Jan 2023 18:29:40 +0000 Subject: 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 Change-Id: I129dfcb8543f8a3a297c0589c841be20ef3b6407 --- src/armnnTfLiteParser/test/FullyConnected.cpp | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'src/armnnTfLiteParser/test/FullyConnected.cpp') 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 }); -- cgit v1.2.1