aboutsummaryrefslogtreecommitdiff
path: root/src/backends/cl/ClWorkloadFactory.cpp
diff options
context:
space:
mode:
authorCathal Corbett <cathal.corbett@arm.com>2022-02-09 23:21:35 +0000
committerCathal Corbett <cathal.corbett@arm.com>2022-02-22 08:46:07 +0000
commitad9171701e6032b3ddf3573f85780bae30c512c6 (patch)
treea5a24cfecc3eb525524d8e63a194b18735176304 /src/backends/cl/ClWorkloadFactory.cpp
parentb0baff73b1574a198e57d46fcd704cedc43cea16 (diff)
downloadarmnn-ad9171701e6032b3ddf3573f85780bae30c512c6.tar.gz
IVGCVSW-6267 Add support of Unidirectional Sequence Lstm fp32/fp16 to Cl
!ComputeLibrary:7150 Signed-off-by: Cathal Corbett <cathal.corbett@arm.com> Change-Id: I01690e6555978d93c41d09bbe5378683bc925f61
Diffstat (limited to 'src/backends/cl/ClWorkloadFactory.cpp')
-rw-r--r--src/backends/cl/ClWorkloadFactory.cpp7
1 files changed, 7 insertions, 0 deletions
diff --git a/src/backends/cl/ClWorkloadFactory.cpp b/src/backends/cl/ClWorkloadFactory.cpp
index 0632787db0..c561bf2157 100644
--- a/src/backends/cl/ClWorkloadFactory.cpp
+++ b/src/backends/cl/ClWorkloadFactory.cpp
@@ -684,6 +684,13 @@ std::unique_ptr<IWorkload> ClWorkloadFactory::CreateWorkload(LayerType type,
m_MemoryManager->GetIntraLayerManager(),
m_CLCompileContext);
}
+ case LayerType::UnidirectionalSequenceLstm :
+ {
+ auto desc = PolymorphicDowncast<const UnidirectionalSequenceLstmQueueDescriptor*>(&descriptor);
+ return MakeWorkloadHelper<ClUnidirectionalSequenceLstmFloatWorkload, NullWorkload>(*desc,
+ info,
+ m_CLCompileContext);
+ }
default:
return nullptr;
}