diff options
| -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); } |