commit f40d6d41b58a086273513d1b386725a7ae9c5503
author Sadik Armagan <sadik.armagan@arm.com> Thu Apr 22 09:12:11 2021 +0100
committer Sadik Armagan <sadik.armagan@arm.com> Thu Apr 22 13:01:26 2021 +0000
tree e21aff17d9e9e86a32c0e0e3e3cb1598a1c20a07
parent bf7edb619cd5cf0ae84342299abe7c27f3ba6e7d

IVGCVSW-5420 'Add CL support for CAST'

Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I1bbf1f2edf7282cb69b99e22de1b8a2fe6e443c2

src/backends/cl/ClLayerSupport.cpp

diff --git a/src/backends/cl/ClLayerSupport.cpp b/src/backends/cl/ClLayerSupport.cpp
index cdbbf0c..aa83826 100644
--- a/src/backends/cl/ClLayerSupport.cpp
+++ b/src/backends/cl/ClLayerSupport.cpp
@@ -25,6 +25,7 @@
 #include "workloads/ClArgMinMaxWorkload.hpp"
 #include "workloads/ClBatchNormalizationFloatWorkload.hpp"
 #include "workloads/ClBatchToSpaceNdWorkload.hpp"
+#include "workloads/ClCastWorkload.hpp"
 #include "workloads/ClComparisonWorkload.hpp"
 #include "workloads/ClConstantWorkload.hpp"
 #include "workloads/ClConvertFp16ToFp32Workload.hpp"
@@ -239,6 +240,16 @@
                                    nullptr);
 }
 
+bool ClLayerSupport::IsCastSupported(const TensorInfo& input,
+                                     const TensorInfo& output,
+                                     Optional<std::string&> reasonIfUnsupported) const
+{
+    FORWARD_WORKLOAD_VALIDATE_FUNC(ClCastValidate,
+                                   reasonIfUnsupported,
+                                   input,
+                                   output);
+}
+
 bool ClLayerSupport::IsBatchToSpaceNdSupported(const TensorInfo& input,
                                                const TensorInfo& output,
                                                const BatchToSpaceNdDescriptor& descriptor,