diff options
author | David Beck <david.beck@arm.com> | 2018-09-07 16:19:24 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-09-25 14:54:29 +0100 |
commit | 4a8692cf18ebd3c4de125274d5c840d7be64e3cd (patch) | |
tree | b504b5f42a83a89c7a40bc3dea13f230c847cc0e /src/armnn/backends/ClLayerSupport.cpp | |
parent | a6bf9121e7c26561ca7cb950020db6cb665596a2 (diff) | |
download | armnn-4a8692cf18ebd3c4de125274d5c840d7be64e3cd.tar.gz |
IVGCVSW-1801 : Cl implementation for SUB
Change-Id: Ia2e1dda8653197454a50679d49020397f5327979
Diffstat (limited to 'src/armnn/backends/ClLayerSupport.cpp')
-rw-r--r-- | src/armnn/backends/ClLayerSupport.cpp | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/src/armnn/backends/ClLayerSupport.cpp b/src/armnn/backends/ClLayerSupport.cpp index 7b5fee2175..3dba1ec94c 100644 --- a/src/armnn/backends/ClLayerSupport.cpp +++ b/src/armnn/backends/ClLayerSupport.cpp @@ -29,6 +29,7 @@ #include "ClWorkloads/ClPermuteWorkload.hpp" #include "ClWorkloads/ClNormalizationFloatWorkload.hpp" #include "ClWorkloads/ClSoftmaxBaseWorkload.hpp" +#include "ClWorkloads/ClSubtractionFloatWorkload.hpp" #include "ClWorkloads/ClLstmFloatWorkload.hpp" #endif @@ -255,8 +256,10 @@ bool IsSubtractionSupportedCl(const TensorInfo& input0, const TensorInfo& output, std::string* reasonIfUnsupported) { - // At the moment subtraction is not supported - return false; + return FORWARD_CL_LAYER_SUPPORT_FUNC(ClSubtractionValidate(input0, + input1, + output, + reasonIfUnsupported)); } bool IsFullyConnectedSupportedCl(const TensorInfo& input, |