diff options
author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-08-06 11:01:53 +0100 |
---|---|---|
committer | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-08-06 11:01:53 +0100 |
commit | c2291fcefde372032aaa5a2b8260801331c76c7c (patch) | |
tree | 829007c8c5a72edefa534e70b54c2edfc1ba6010 /src | |
parent | ff9cb791eb2c6c67ace5b983f38772568e38d2c4 (diff) | |
download | armnn-c2291fcefde372032aaa5a2b8260801331c76c7c.tar.gz |
GitHub #240 Fix bias shape in CreatePreCompiledWorkloadTest()
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: I10b8d5e2e95eb8f48a6be29005136459c3da2957
Diffstat (limited to 'src')
-rw-r--r-- | src/armnn/test/CreateWorkload.hpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/armnn/test/CreateWorkload.hpp b/src/armnn/test/CreateWorkload.hpp index b576f12c22..b50135aea5 100644 --- a/src/armnn/test/CreateWorkload.hpp +++ b/src/armnn/test/CreateWorkload.hpp @@ -1334,7 +1334,7 @@ std::pair<armnn::IOptimizedNetworkPtr, std::unique_ptr<PreCompiledWorkload>> Cre constexpr armnn::DataType biasDataType = ( dataType == armnn::DataType::QuantisedAsymm8) ? armnn::DataType::Signed32 : armnn::DataType::Float32; - TensorInfo biasTensorInfo(TensorShape({1, 1, 1, 16}), biasDataType, 0.9f * 0.9f, 0); + TensorInfo biasTensorInfo(TensorShape({16}), biasDataType, 0.9f * 0.9f, 0); unsigned int biasLength = biasTensorInfo.GetNumElements(); using BiasType = armnn::ResolveType<biasDataType>; |