diff options
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, |