commit 8398edcfb933b638ddf4b88d84d6e188c49b1e0d
author Teresa Charlin <teresa.charlinreyes@arm.com> Mon Jul 20 14:23:02 2020 +0100
committer Jim Flynn <jim.flynn@arm.com> Fri Jul 24 22:51:19 2020 +0000
tree 2dbb158b19c135007baf6178d84acf9e770938d9
parent 2ebe095eec687f03c2f23051b31cfa4309d3e67b

IVGCVSW-3896 Add CL LOG_SOFTMAX Workload

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

src/backends/cl/ClLayerSupport.cpp

diff --git a/src/backends/cl/ClLayerSupport.cpp b/src/backends/cl/ClLayerSupport.cpp
index 0bff963..74b5aad 100644
--- a/src/backends/cl/ClLayerSupport.cpp
+++ b/src/backends/cl/ClLayerSupport.cpp
@@ -1,5 +1,5 @@
 //
-// Copyright © 2017 Arm Ltd. All rights reserved.
+// Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
 // SPDX-License-Identifier: MIT
 //
 
@@ -38,6 +38,7 @@
 #include "workloads/ClGatherWorkload.hpp"
 #include "workloads/ClInstanceNormalizationWorkload.hpp"
 #include "workloads/ClL2NormalizationFloatWorkload.hpp"
+#include "workloads/ClLogSoftmaxWorkload.hpp"
 #include "workloads/ClLstmFloatWorkload.hpp"
 #include "workloads/ClMaximumWorkload.hpp"
 #include "workloads/ClMeanWorkload.hpp"
@@ -516,6 +517,18 @@
                                    descriptor);
 }
 
+bool ClLayerSupport::IsLogSoftmaxSupported(const TensorInfo& input,
+                                                const TensorInfo& output,
+                                                const LogSoftmaxDescriptor& descriptor,
+                                                Optional<std::string&> reasonIfUnsupported) const
+{
+    FORWARD_WORKLOAD_VALIDATE_FUNC(ClLogSoftmaxWorkloadValidate,
+                                   reasonIfUnsupported,
+                                   input,
+                                   output,
+                                   descriptor);
+}
+
 bool ClLayerSupport::IsLstmSupported(const TensorInfo& input,
                                      const TensorInfo& outputStateIn,
                                      const TensorInfo& cellStateIn,