aboutsummaryrefslogtreecommitdiff
path: root/src/backends/neon/NeonWorkloadFactory.hpp
diff options
context:
space:
mode:
authorTeresa Charlin <teresa.charlinreyes@arm.com>2021-10-22 17:15:00 +0100
committerTeresaARM <teresa.charlinreyes@arm.com>2021-10-27 09:32:11 +0000
commitec5f7d13582d1e477dc3473223b503388092a352 (patch)
tree37c4d6299f0726a853e28f6297d4eb222cda2c68 /src/backends/neon/NeonWorkloadFactory.hpp
parente1fdd2866b0f403b5e80994890d62c2c038c16c9 (diff)
downloadarmnn-ec5f7d13582d1e477dc3473223b503388092a352.tar.gz
IVGCVSW-6170 Add CpuAcc Conv3d Workload
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com> Change-Id: I3a5bfef5a0085d172fd3689e67f25af909ace2ee
Diffstat (limited to 'src/backends/neon/NeonWorkloadFactory.hpp')
-rw-r--r--src/backends/neon/NeonWorkloadFactory.hpp3
1 files changed, 3 insertions, 0 deletions
diff --git a/src/backends/neon/NeonWorkloadFactory.hpp b/src/backends/neon/NeonWorkloadFactory.hpp
index bd84c057f5..f44681a5a8 100644
--- a/src/backends/neon/NeonWorkloadFactory.hpp
+++ b/src/backends/neon/NeonWorkloadFactory.hpp
@@ -97,6 +97,9 @@ public:
std::unique_ptr<IWorkload> CreateConvolution2d(const Convolution2dQueueDescriptor& descriptor,
const WorkloadInfo& info) const override;
+ std::unique_ptr<IWorkload> CreateConvolution3d(const Convolution3dQueueDescriptor& descriptor,
+ const WorkloadInfo& info) const override;
+
std::unique_ptr<IWorkload> CreateDebug(const DebugQueueDescriptor& descriptor,
const WorkloadInfo& info) const override;