commit d5166106ebfc11b1db832a708892e12edeaf997e
author Matthew Jackson <matthew.jackson@arm.com> Wed Jul 31 14:06:28 2019 +0100
committer Matteo Martincigh <matteo.martincigh@arm.com> Thu Aug 01 09:42:03 2019 +0000
tree 8f4be789711ed50b0bddf50a813caf4b823a59e9
parent 2208b604dea6919525a12779df9aa5e49ea6a539

IVGCVSW-3539 Add CL support and tests for Stack

* Added CL backend support for Stack
* Added unit tests for Stack on the CL backend
* Refactored unit tests to support generic data types

Signed-off-by: Matthew Jackson <matthew.jackson@arm.com>
Change-Id: I38ee3e9d8947ea98a3104c982698001e704d7d89

src/backends/cl/ClLayerSupport.cpp

diff --git a/src/backends/cl/ClLayerSupport.cpp b/src/backends/cl/ClLayerSupport.cpp
index b737daf..0553962 100644
--- a/src/backends/cl/ClLayerSupport.cpp
+++ b/src/backends/cl/ClLayerSupport.cpp
@@ -46,6 +46,7 @@
 #include "workloads/ClSpaceToBatchNdWorkload.hpp"
 #include "workloads/ClSpaceToDepthWorkload.hpp"
 #include "workloads/ClSplitterWorkload.hpp"
+#include "workloads/ClStackWorkload.hpp"
 #include "workloads/ClStridedSliceWorkload.hpp"
 #include "workloads/ClSubtractionWorkload.hpp"
 #include "workloads/ClTransposeConvolution2dWorkload.hpp"
@@ -671,6 +672,18 @@
     return true;
 }
 
+bool ClLayerSupport::IsStackSupported(const std::vector<const TensorInfo*>& inputs,
+                                      const TensorInfo& output,
+                                      const StackDescriptor& descriptor,
+                                      Optional<std::string&> reasonIfUnsupported) const
+{
+    FORWARD_WORKLOAD_VALIDATE_FUNC(ClStackWorkloadValidate,
+                                   reasonIfUnsupported,
+                                   inputs,
+                                   output,
+                                   descriptor);
+}
+
 bool ClLayerSupport::IsStridedSliceSupported(const TensorInfo& input,
                                              const TensorInfo& output,
                                              const StridedSliceDescriptor& descriptor,