From 581742d8b6c25bf0917d6bdcd5f9b8a5e72b370d Mon Sep 17 00:00:00 2001 From: Sadik Armagan Date: Mon, 12 Aug 2019 14:11:37 +0100 Subject: IVGCVSW-3163 Support Neon TransposeConv2d * NEON support for TransposeConolution2d * Indentation fix on SpaceToDepth LayerSupport method Signed-off-by: Keith Davis Signed-off-by: Sadik Armagan Signed-off-by: Aron Virginas-Tar Change-Id: Ie3a01a22e05f62f475f768dfe05ce561d2f3b0ad --- src/backends/neon/NeonWorkloadFactory.cpp | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'src/backends/neon/NeonWorkloadFactory.cpp') diff --git a/src/backends/neon/NeonWorkloadFactory.cpp b/src/backends/neon/NeonWorkloadFactory.cpp index 36b176626d..a22a64fe91 100644 --- a/src/backends/neon/NeonWorkloadFactory.cpp +++ b/src/backends/neon/NeonWorkloadFactory.cpp @@ -398,6 +398,14 @@ std::unique_ptr NeonWorkloadFactory::CreateRsqrt(const RsqrtQueueDesc return MakeWorkloadHelper(descriptor, info); } +std::unique_ptr NeonWorkloadFactory::CreateTransposeConvolution2d( + const TransposeConvolution2dQueueDescriptor &descriptor, + const WorkloadInfo &info) const +{ + return std::make_unique(descriptor, info, + m_MemoryManager->GetIntraLayerManager()); +} + std::unique_ptr NeonWorkloadFactory::CreatePreCompiled(const PreCompiledQueueDescriptor& descriptor, const WorkloadInfo& info) const { -- cgit v1.2.1