aboutsummaryrefslogtreecommitdiff
path: root/src/backends/neon/workloads/NeonSpaceToDepthWorkload.hpp
diff options
context:
space:
mode:
authorEllen Norris-Thompson <ellen.norris-thompson@arm.com>2019-06-26 16:40:36 +0100
committerJames Conroy <james.conroy@arm.com>2019-07-09 13:29:58 +0000
commit2979457c5974e298033c387a5cd33f95e3301623 (patch)
treedf7b6b54def6c608602f487e4cffa0b1dc187a3f /src/backends/neon/workloads/NeonSpaceToDepthWorkload.hpp
parent39f4639a79625c8f37c6ca547dadc7925378ee3e (diff)
downloadarmnn-2979457c5974e298033c387a5cd33f95e3301623.tar.gz
IVGCVSW-3355 Add NeonSpaceToDepthWorkload
* Add Neon backend support for SpaceToDepth * Enabled Neon Unit, CreateWorkload and EndToEnd tests for SpaceToDepth. * Added QSymm16 support to NeonTensorHandle. Signed-off-by: Ellen Norris-Thompson <ellen.norris-thompson@arm.com> Change-Id: Ibbb0bce5ad77dace2bf4c4c111006d2fadf844e8
Diffstat (limited to 'src/backends/neon/workloads/NeonSpaceToDepthWorkload.hpp')
-rw-r--r--src/backends/neon/workloads/NeonSpaceToDepthWorkload.hpp31
1 files changed, 31 insertions, 0 deletions
diff --git a/src/backends/neon/workloads/NeonSpaceToDepthWorkload.hpp b/src/backends/neon/workloads/NeonSpaceToDepthWorkload.hpp
new file mode 100644
index 0000000000..f2de26b3c6
--- /dev/null
+++ b/src/backends/neon/workloads/NeonSpaceToDepthWorkload.hpp
@@ -0,0 +1,31 @@
+//
+// Copyright © 2017 Arm Ltd. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#pragma once
+
+#include <armnn/TypesUtils.hpp>
+
+#include <backendsCommon/Workload.hpp>
+
+#include <arm_compute/runtime/NEON/functions/NESpaceToDepthLayer.h>
+
+namespace armnn
+{
+
+arm_compute::Status NeonSpaceToDepthWorkloadValidate(const TensorInfo& input,
+ const TensorInfo& output,
+ const SpaceToDepthDescriptor& descriptor);
+
+class NeonSpaceToDepthWorkload : public BaseWorkload<SpaceToDepthQueueDescriptor>
+{
+public:
+ using BaseWorkload<SpaceToDepthQueueDescriptor>::BaseWorkload;
+ NeonSpaceToDepthWorkload(const SpaceToDepthQueueDescriptor& descriptor, const WorkloadInfo& info);
+ virtual void Execute() const override;
+private:
+ mutable std::unique_ptr<arm_compute::NESpaceToDepthLayer> m_Layer;
+};
+
+} //namespace armnn \ No newline at end of file