aboutsummaryrefslogtreecommitdiff
path: root/src/backends/neon/NeonWorkloadFactory.cpp
diff options
context:
space:
mode:
authorRyan OShea <ryan.oshea3@arm.com>2022-05-04 00:38:03 +0100
committerryan.oshea3 <ryan.oshea3@arm.com>2022-05-13 16:46:20 +0000
commit19e79420fa9f136352cce56b5650b47aa2e96de3 (patch)
treee6f0eb1907377ff78b6a905d028f394d7faa880e /src/backends/neon/NeonWorkloadFactory.cpp
parent4452baf3d295164877c5810a3867b1d2d79b04f3 (diff)
downloadarmnn-19e79420fa9f136352cce56b5650b47aa2e96de3.tar.gz
IVGCVSW-6175 Add Pooling3d to Neon
* Add IsSupported for Pooling3d * Add CreateWorkload case for Pooling3d * Create new NeonPooling3dWorkload header and source files * Add Pooling3d workload to NeonWorkloads.hpp * Add float32 tests for Pooling3d workload * Add Uint8 tests for Cl and NE pooling3d Signed-off-by: Ryan OShea <ryan.oshea3@arm.com> Change-Id: Ic992e1233d1eb8db52df2c8446183df1c907bc4d
Diffstat (limited to 'src/backends/neon/NeonWorkloadFactory.cpp')
-rw-r--r--src/backends/neon/NeonWorkloadFactory.cpp5
1 files changed, 5 insertions, 0 deletions
diff --git a/src/backends/neon/NeonWorkloadFactory.cpp b/src/backends/neon/NeonWorkloadFactory.cpp
index f087c6262a..ff9ef268ec 100644
--- a/src/backends/neon/NeonWorkloadFactory.cpp
+++ b/src/backends/neon/NeonWorkloadFactory.cpp
@@ -461,6 +461,11 @@ std::unique_ptr<IWorkload> NeonWorkloadFactory::CreateWorkload(LayerType type,
auto pooling2dQueueDescriptor = PolymorphicDowncast<const Pooling2dQueueDescriptor*>(&descriptor);
return std::make_unique<NeonPooling2dWorkload>(*pooling2dQueueDescriptor, info);
}
+ case LayerType::Pooling3d :
+ {
+ auto pooling3dQueueDescriptor = PolymorphicDowncast<const Pooling3dQueueDescriptor*>(&descriptor);
+ return std::make_unique<NeonPooling3dWorkload>(*pooling3dQueueDescriptor, info);
+ }
case LayerType::PreCompiled :
{
auto preCompiledQueueDescriptor = PolymorphicDowncast<const PreCompiledQueueDescriptor*>(&descriptor);