From c2291fcefde372032aaa5a2b8260801331c76c7c Mon Sep 17 00:00:00 2001
From: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Date: Tue, 6 Aug 2019 11:01:53 +0100
Subject: GitHub #240 Fix bias shape in CreatePreCompiledWorkloadTest()

Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: I10b8d5e2e95eb8f48a6be29005136459c3da2957
---
 src/armnn/test/CreateWorkload.hpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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>;
-- 
cgit v1.2.1