diff options
| author | David Beck <david.beck@arm.com> | 2018-09-10 14:47:28 +0100 |
|---|---|---|
| committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-01 14:56:48 +0100 |
| commit | bc3924503cf7838eea7e9fcca4a22e2bf54ba97f (patch) | |
| tree | 6ea56e796bef35ffa9397eedad956363486241c7 /src/armnn/backends/NeonLayerSupport.cpp | |
| parent | 32b9046ea74d2387a08819cf5e67c183e03f6d3f (diff) | |
| download | armnn-bc3924503cf7838eea7e9fcca4a22e2bf54ba97f.tar.gz | |
IVGCVSW-1802 : Neon Sub Workload
Change-Id: I1e914b046cd8fd25669390d477f101098fe0d476
Diffstat (limited to 'src/armnn/backends/NeonLayerSupport.cpp')
| -rw-r--r-- | src/armnn/backends/NeonLayerSupport.cpp | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/src/armnn/backends/NeonLayerSupport.cpp b/src/armnn/backends/NeonLayerSupport.cpp index 7f33c48ed1..30956dfba0 100644 --- a/src/armnn/backends/NeonLayerSupport.cpp +++ b/src/armnn/backends/NeonLayerSupport.cpp @@ -27,6 +27,7 @@ #include "NeonWorkloads/NeonPermuteWorkload.hpp" #include "NeonWorkloads/NeonPooling2dBaseWorkload.hpp" #include "NeonWorkloads/NeonSoftmaxBaseWorkload.hpp" +#include "NeonWorkloads/NeonSubtractionFloatWorkload.hpp" #endif using namespace boost; @@ -239,8 +240,11 @@ bool IsSubtractionSupportedNeon(const TensorInfo& input0, const TensorInfo& output, std::string* reasonIfUnsupported) { - // At the moment subtraction is not supported - return false; + FORWARD_WORKLOAD_VALIDATE_FUNC(NeonSubtractionWorkloadValidate, + reasonIfUnsupported, + input0, + input1, + output); } bool IsFullyConnectedSupportedNeon(const TensorInfo& input, |