diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2020-05-22 08:36:33 +0100 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2020-05-22 11:50:20 +0000 |
commit | d6f06496e41061d74abb3548374e5b52a2670bc3 (patch) | |
tree | 190a73351fa540012cfbab8dc92f21da9e018ddf /src/backends | |
parent | a7227ac8fa45c9ea0da0e1ed66bb0c551c61095b (diff) | |
download | armnn-d6f06496e41061d74abb3548374e5b52a2670bc3.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
Diffstat (limited to 'src/backends')
-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); } |