aboutsummaryrefslogtreecommitdiff
path: root/src/backends/neon/workloads/NeonResizeWorkload.hpp
diff options
context:
space:
mode:
authorEllen Norris-Thompson <ellen.norris-thompson@arm.com>2019-07-15 14:23:30 +0100
committerNarumol Prangnawarat <narumol.prangnawarat@arm.com>2019-07-17 16:36:30 +0000
commit37e68686a4889b6e79b035356202903647671f13 (patch)
treef248e6ba83a3dcecc0c0bef829123a4378c23e99 /src/backends/neon/workloads/NeonResizeWorkload.hpp
parentbcca1f4a7ba8364f7b5e58e8e8866ccd7d530f92 (diff)
downloadarmnn-37e68686a4889b6e79b035356202903647671f13.tar.gz
IVGCVSW-3297 Add Neon backend support for ResizeNearestNeighbour
* Renamed NeonResizeBilinearWorkload.* to NeonResizeWorkload.* and added support for ResizeNearestNeighbour * Added CreateWorkload and LayerTests for Neon backend Signed-off-by: Ellen Norris-Thompson <ellen.norris-thompson@arm.com> Change-Id: I72f5340608a0928f8b32a41d1915ee2c35849f18
Diffstat (limited to 'src/backends/neon/workloads/NeonResizeWorkload.hpp')
-rw-r--r--src/backends/neon/workloads/NeonResizeWorkload.hpp29
1 files changed, 29 insertions, 0 deletions
diff --git a/src/backends/neon/workloads/NeonResizeWorkload.hpp b/src/backends/neon/workloads/NeonResizeWorkload.hpp
new file mode 100644
index 0000000000..69995c6311
--- /dev/null
+++ b/src/backends/neon/workloads/NeonResizeWorkload.hpp
@@ -0,0 +1,29 @@
+//
+// Copyright © 2017 Arm Ltd. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#pragma once
+
+#include <backendsCommon/Workload.hpp>
+
+#include <arm_compute/runtime/NEON/functions/NEScale.h>
+
+namespace armnn
+{
+
+arm_compute::Status NeonResizeWorkloadValidate(const TensorInfo& input,
+ const TensorInfo& output,
+ const ResizeDescriptor& descriptor);
+
+class NeonResizeWorkload : public BaseWorkload<ResizeQueueDescriptor>
+{
+public:
+ NeonResizeWorkload(const ResizeQueueDescriptor& descriptor, const WorkloadInfo& info);
+ void Execute() const override;
+
+private:
+ mutable arm_compute::NEScale m_ResizeLayer;
+};
+
+} //namespace armnn