commit d6f06496e41061d74abb3548374e5b52a2670bc3
author Sadik Armagan <sadik.armagan@arm.com> Fri May 22 08:36:33 2020 +0100
committer Sadik Armagan <sadik.armagan@arm.com> Fri May 22 11:50:20 2020 +0000
tree 190a73351fa540012cfbab8dc92f21da9e018ddf
parent a7227ac8fa45c9ea0da0e1ed66bb0c551c61095b

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

diff --git a/src/backends/backendsCommon/WorkloadData.cpp b/src/backends/backendsCommon/WorkloadData.cpp
index 5796fc7..8f751c4 100644
--- a/src/backends/backendsCommon/WorkloadData.cpp
+++ b/src/backends/backendsCommon/WorkloadData.cpp
@@ -3116,7 +3116,7 @@
         if (m_ProjectionBias)
         {
             auto projectionBiasInfo = m_ProjectionBias->GetTensorInfo();
-            ValidateTensorNumDimNumElem(projectionBiasInfo, 1, numUnits, "ProjectionBias");
+            ValidateTensorNumDimNumElem(projectionBiasInfo, 1, outputSize, "ProjectionBias");
             ValidateDataTypes(projectionBiasInfo, biasSupportedTypes, descriptorName);
         }