commit a57eccbe313557b9eafec40b39bac3115d9b930d
author Keith Davis <keith.davis@arm.com> Fri Jun 14 17:33:22 2019 +0100
committer Keith Davis <keith.davis@arm.com> Wed Jun 19 14:43:26 2019 +0100
tree 2522157537f46fa2e54c0405d9e9e60f5f0a9ae1
parent ab9e52563f624d9782b97400f643d2632cc8d770

IVGCVSW-3231 Add reference workload support for SpaceToDepth

* Added reference workload for SpaceToDepth
* Added unit tests for float32 & uint8
* Minor sort refactoring to RefWorkloads.hpp to alphabetical order

Change-Id: I2e01f8101650e2aae102a8a32bc0064f067141ab
Signed-off-by: Keith Davis <keith.davis@arm.com>

src/backends/reference/RefLayerSupport.hpp

diff --git a/src/backends/reference/RefLayerSupport.hpp b/src/backends/reference/RefLayerSupport.hpp
index 041701d..5d24149 100644
--- a/src/backends/reference/RefLayerSupport.hpp
+++ b/src/backends/reference/RefLayerSupport.hpp
@@ -236,6 +236,12 @@
                                    const SpaceToBatchNdDescriptor& descriptor,
                                    Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
 
+    bool IsSpaceToDepthSupported(const TensorInfo& input,
+                                 const TensorInfo& output,
+                                 const SpaceToDepthDescriptor& descriptor,
+                                 Optional<std::string&> reasonIfUnsupported = EmptyOptional())
+                                 const override;
+
     ARMNN_DEPRECATED_MSG("Use IsSplitterSupported with outputs instead")
     bool IsSplitterSupported(const TensorInfo& input,
                              const ViewsDescriptor& descriptor,