diff options
| author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-09-09 13:36:45 +0100 |
|---|---|---|
| committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-09-10 08:59:25 +0000 |
| commit | 914e4db5a9083e922d89f133672fd44e92016e96 (patch) | |
| tree | fd3d7d4a5c739eef02c756a668fa4a95c2255f36 /src/backends/neon/NeonLayerSupport.cpp | |
| parent | ab173e9b6978d5befb4884a803773967d52bcfef (diff) | |
| download | armnn-914e4db5a9083e922d89f133672fd44e92016e96.tar.gz | |
IVGCVSW-3822 Add NEON workload support for ABS
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: I0c2ead004200e053d748ea39937f2f9ed35a636b
Diffstat (limited to 'src/backends/neon/NeonLayerSupport.cpp')
| -rw-r--r-- | src/backends/neon/NeonLayerSupport.cpp | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/src/backends/neon/NeonLayerSupport.cpp b/src/backends/neon/NeonLayerSupport.cpp index 80695fb3c7..67c156001d 100644 --- a/src/backends/neon/NeonLayerSupport.cpp +++ b/src/backends/neon/NeonLayerSupport.cpp @@ -18,6 +18,7 @@ #if defined(ARMCOMPUTENEON_ENABLED) #include <aclCommon/ArmComputeUtils.hpp> +#include "workloads/NeonAbsWorkload.hpp" #include "workloads/NeonAdditionWorkload.hpp" #include "workloads/NeonActivationWorkload.hpp" #include "workloads/NeonBatchNormalizationWorkload.hpp" @@ -108,6 +109,16 @@ inline bool IsWorkloadSupported(FuncType& func, Optional<std::string&> reasonIfU } // anonymous namespace +bool NeonLayerSupport::IsAbsSupported(const TensorInfo& input, + const TensorInfo& output, + Optional<std::string&> reasonIfUnsupported) const +{ + FORWARD_WORKLOAD_VALIDATE_FUNC(NeonAbsWorkloadValidate, + reasonIfUnsupported, + input, + output); +} + bool NeonLayerSupport::IsActivationSupported(const TensorInfo& input, const TensorInfo& output, const ActivationDescriptor& descriptor, |