commit b280196ef4e04725a46155732ff0a1faf6b76722
author Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> Mon Sep 30 11:24:53 2019 +0100
committer Kevin May <kevin.may@arm.com> Mon Sep 30 16:53:23 2019 +0000
tree f63b0016715003d9842f264123b455afb515c6a5
parent 2f00b7421f259851448a0be72e9eb01141faad2e

IVGCVSW-3732 Add CL workload for DepthToSpace

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

src/backends/cl/ClLayerSupport.cpp

diff --git a/src/backends/cl/ClLayerSupport.cpp b/src/backends/cl/ClLayerSupport.cpp
index 5905cd9..4503915 100644
--- a/src/backends/cl/ClLayerSupport.cpp
+++ b/src/backends/cl/ClLayerSupport.cpp
@@ -24,8 +24,9 @@
 #include "workloads/ClConvertFp16ToFp32Workload.hpp"
 #include "workloads/ClConvertFp32ToFp16Workload.hpp"
 #include "workloads/ClConvolution2dWorkload.hpp"
-#include "workloads/ClDequantizeWorkload.hpp"
+#include "workloads/ClDepthToSpaceWorkload.hpp"
 #include "workloads/ClDepthwiseConvolutionWorkload.hpp"
+#include "workloads/ClDequantizeWorkload.hpp"
 #include "workloads/ClDivisionFloatWorkload.hpp"
 #include "workloads/ClFullyConnectedWorkload.hpp"
 #include "workloads/ClGreaterWorkload.hpp"
@@ -303,6 +304,18 @@
                                    output);
 }
 
+bool ClLayerSupport::IsDepthToSpaceSupported(const TensorInfo& input,
+                                             const TensorInfo& output,
+                                             const DepthToSpaceDescriptor& descriptor,
+                                             Optional<std::string&> reasonIfUnsupported) const
+{
+    FORWARD_WORKLOAD_VALIDATE_FUNC(ClDepthToSpaceWorkloadValidate,
+                                   reasonIfUnsupported,
+                                   input,
+                                   output,
+                                   descriptor);
+}
+
 bool ClLayerSupport::IsDepthwiseConvolutionSupported(const TensorInfo& input,
                                                      const TensorInfo& output,
                                                      const DepthwiseConvolution2dDescriptor& descriptor,