From 9f008f8aaff25946b83d55b61fa6070488540665 Mon Sep 17 00:00:00 2001
From: Sadik Armagan <sadik.armagan@arm.com>
Date: Fri, 22 May 2020 08:36:33 +0100
Subject: IVGCVSW-4453 QLSTM Support

* Number of elements is output size for projection bias on QLSTM

Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I9d32cfb187bbe0c6ef809a7a89da907fbac83380
---
 src/backends/backendsCommon/WorkloadData.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/backends/backendsCommon/WorkloadData.cpp b/src/backends/backendsCommon/WorkloadData.cpp
index 5796fc7c77..8f751c432f 100644
--- a/src/backends/backendsCommon/WorkloadData.cpp
+++ b/src/backends/backendsCommon/WorkloadData.cpp
@@ -3116,7 +3116,7 @@ void QLstmQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) const
         if (m_ProjectionBias)
         {
             auto projectionBiasInfo = m_ProjectionBias->GetTensorInfo();
-            ValidateTensorNumDimNumElem(projectionBiasInfo, 1, numUnits, "ProjectionBias");
+            ValidateTensorNumDimNumElem(projectionBiasInfo, 1, outputSize, "ProjectionBias");
             ValidateDataTypes(projectionBiasInfo, biasSupportedTypes, descriptorName);
         }
 
-- 
cgit v1.2.1