aboutsummaryrefslogtreecommitdiff
path: root/src/backends/backendsCommon/WorkloadFactory.cpp
diff options
context:
space:
mode:
authorJames Conroy <james.conroy@arm.com>2020-05-18 15:16:42 +0100
committerJames Conroy <james.conroy@arm.com>2020-05-18 15:22:15 +0100
commited324051794a262a7694acfb0a378724d6ba8401 (patch)
tree32d516998be61052492ef1555d778615ee489412 /src/backends/backendsCommon/WorkloadFactory.cpp
parent63b234c6f5dd1340d8a7c5f64d7e1bbbe3e2d8df (diff)
downloadarmnn-ed324051794a262a7694acfb0a378724d6ba8401.tar.gz
IVGCVSW-4449 Add missing QLstm nullptr checks
* Adds missing nullptr checks for peephole bias for QLstm. Signed-off-by: James Conroy <james.conroy@arm.com> Change-Id: Ib04c8352141977cc7ea11a4859f1b11d46472722
Diffstat (limited to 'src/backends/backendsCommon/WorkloadFactory.cpp')
-rw-r--r--src/backends/backendsCommon/WorkloadFactory.cpp7
1 files changed, 6 insertions, 1 deletions
diff --git a/src/backends/backendsCommon/WorkloadFactory.cpp b/src/backends/backendsCommon/WorkloadFactory.cpp
index c55c70a1f7..34bfd7cead 100644
--- a/src/backends/backendsCommon/WorkloadFactory.cpp
+++ b/src/backends/backendsCommon/WorkloadFactory.cpp
@@ -795,7 +795,12 @@ bool IWorkloadFactory::IsLayerSupported(const BackendId& backendId,
if(descriptor.m_ProjectionEnabled)
{
paramsInfo.m_ProjectionWeights = &cLayer->m_ProjectionParameters.m_ProjectionWeights->GetTensorInfo();
- paramsInfo.m_ProjectionBias = &cLayer->m_ProjectionParameters.m_ProjectionBias->GetTensorInfo();
+
+ // Projection bias is optional even if projection is enabled
+ if (cLayer->m_ProjectionParameters.m_ProjectionBias != nullptr)
+ {
+ paramsInfo.m_ProjectionBias = &cLayer->m_ProjectionParameters.m_ProjectionBias->GetTensorInfo();
+ }
}
if(descriptor.m_PeepholeEnabled)