diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2020-05-22 08:36:33 +0100 |
---|---|---|
committer | Jan Eilers <jan.eilers@arm.com> | 2020-05-22 14:43:17 +0100 |
commit | 9f008f8aaff25946b83d55b61fa6070488540665 (patch) | |
tree | b66b6df0189443959065eb9e764210ad52e2d6f4 | |
parent | 72b415674e7df9152a1e5d206b70e4d7a663f3fc (diff) | |
download | armnn-9f008f8aaff25946b83d55b61fa6070488540665.tar.gz |
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
-rw-r--r-- | src/backends/backendsCommon/WorkloadData.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
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); } |