diff options
author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2020-07-20 14:23:02 +0100 |
---|---|---|
committer | Jim Flynn <jim.flynn@arm.com> | 2020-07-24 22:51:19 +0000 |
commit | 8398edcfb933b638ddf4b88d84d6e188c49b1e0d (patch) | |
tree | 2dbb158b19c135007baf6178d84acf9e770938d9 /src/backends/cl/ClLayerSupport.cpp | |
parent | 2ebe095eec687f03c2f23051b31cfa4309d3e67b (diff) | |
download | armnn-8398edcfb933b638ddf4b88d84d6e188c49b1e0d.tar.gz |
IVGCVSW-3896 Add CL LOG_SOFTMAX Workload
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I6c5704c6c7bcf1dd008061c037c81a712365aa74
Diffstat (limited to 'src/backends/cl/ClLayerSupport.cpp')
-rw-r--r-- | src/backends/cl/ClLayerSupport.cpp | 15 |
1 files changed, 14 insertions, 1 deletions
diff --git a/src/backends/cl/ClLayerSupport.cpp b/src/backends/cl/ClLayerSupport.cpp index 0bff96345a..74b5aadb03 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 @@ bool ClLayerSupport::IsL2NormalizationSupported(const TensorInfo& input, 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, |