commit 1222dbd3ddc4e819f8d1b204535bfd222cd1eadd
author Teresa Charlin <teresa.charlinreyes@arm.com> Thu Sep 02 13:58:52 2021 +0100
committer Jim Flynn <jim.flynn@arm.com> Tue Sep 28 21:59:08 2021 +0000
tree f603e5140f1ea19c193a0b6a67c1967f0a24d126
parent ef72938147b75cc37a757d4c9926503435c8e1d5

IVGCVSW-3706 Add Channel Shuffle Workload to GpuAcc backend

Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: Ic37fc49a97a5ca570a23bc415d15ee3841534336

src/backends/cl/ClLayerSupport.cpp

diff --git a/src/backends/cl/ClLayerSupport.cpp b/src/backends/cl/ClLayerSupport.cpp
index f9848ff..0873021 100644
--- a/src/backends/cl/ClLayerSupport.cpp
+++ b/src/backends/cl/ClLayerSupport.cpp
@@ -26,6 +26,7 @@
 #include "workloads/ClBatchNormalizationFloatWorkload.hpp"
 #include "workloads/ClBatchToSpaceNdWorkload.hpp"
 #include "workloads/ClCastWorkload.hpp"
+#include "workloads/ClChannelShuffleWorkload.hpp"
 #include "workloads/ClComparisonWorkload.hpp"
 #include "workloads/ClConstantWorkload.hpp"
 #include "workloads/ClConvertFp16ToFp32Workload.hpp"
@@ -242,6 +243,18 @@
                                    nullptr);
 }
 
+bool ClLayerSupport::IsBatchToSpaceNdSupported(const TensorInfo& input,
+                                               const TensorInfo& output,
+                                               const BatchToSpaceNdDescriptor& descriptor,
+                                               Optional<std::string&> reasonIfUnsupported) const
+{
+    FORWARD_WORKLOAD_VALIDATE_FUNC(ClBatchToSpaceNdWorkloadValidate,
+                                   reasonIfUnsupported,
+                                   input,
+                                   output,
+                                   descriptor);
+}
+
 bool ClLayerSupport::IsCastSupported(const TensorInfo& input,
                                      const TensorInfo& output,
                                      Optional<std::string&> reasonIfUnsupported) const
@@ -252,12 +265,12 @@
                                    output);
 }
 
-bool ClLayerSupport::IsBatchToSpaceNdSupported(const TensorInfo& input,
+bool ClLayerSupport::IsChannelShuffleSupported(const TensorInfo& input,
                                                const TensorInfo& output,
-                                               const BatchToSpaceNdDescriptor& descriptor,
+                                               const ChannelShuffleDescriptor& descriptor,
                                                Optional<std::string&> reasonIfUnsupported) const
 {
-    FORWARD_WORKLOAD_VALIDATE_FUNC(ClBatchToSpaceNdWorkloadValidate,
+    FORWARD_WORKLOAD_VALIDATE_FUNC(ClChannelShuffleValidate,
                                    reasonIfUnsupported,
                                    input,
                                    output,