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

diff --git a/src/backends/backendsCommon/WorkloadData.cpp b/src/backends/backendsCommon/WorkloadData.cpp
index 5796fc7..8f751c4 100644 (file)
--- 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();
         if (m_ProjectionBias)
         {
             auto projectionBiasInfo = m_ProjectionBias->GetTensorInfo();

-            ValidateTensorNumDimNumElem(projectionBiasInfo, 1, numUnits, "ProjectionBias");
+            ValidateTensorNumDimNumElem(projectionBiasInfo, 1, outputSize, "ProjectionBias");

             ValidateDataTypes(projectionBiasInfo, biasSupportedTypes, descriptorName);
         }
 
             ValidateDataTypes(projectionBiasInfo, biasSupportedTypes, descriptorName);
         }