commit 2f00b7421f259851448a0be72e9eb01141faad2e
author Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> Mon Sep 30 13:28:08 2019 +0100
committer Kevin May <kevin.may@arm.com> Mon Sep 30 14:13:42 2019 +0000
tree 109902951005eca1b6bac148c7b41387e487432d
parent 7be47efac07b6276f02a17cb486f9061a426a837

IVGCVSW-3734 Add NEON workload for DepthToSpace

Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: I4fa8cba57f3a2277112c02062f4d2790089d1eb5

src/backends/neon/NeonLayerSupport.hpp

diff --git a/src/backends/neon/NeonLayerSupport.hpp b/src/backends/neon/NeonLayerSupport.hpp
index cc4798e..31f9e57 100644
--- a/src/backends/neon/NeonLayerSupport.hpp
+++ b/src/backends/neon/NeonLayerSupport.hpp
@@ -58,6 +58,12 @@
                                   const Optional<TensorInfo>& biases,
                                   Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
 
+    bool IsDepthToSpaceSupported(const TensorInfo& input,
+                                 const TensorInfo& output,
+                                 const DepthToSpaceDescriptor& descriptor,
+                                 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
+
+
     bool IsDepthwiseConvolutionSupported(const TensorInfo& input,
                                          const TensorInfo& output,
                                          const DepthwiseConvolution2dDescriptor& descriptor,