diff options
Diffstat (limited to 'src/armnn/backends/RefLayerSupport.cpp')
| -rw-r--r-- | src/armnn/backends/RefLayerSupport.cpp | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/src/armnn/backends/RefLayerSupport.cpp b/src/armnn/backends/RefLayerSupport.cpp index 5437574789..41f57f1677 100644 --- a/src/armnn/backends/RefLayerSupport.cpp +++ b/src/armnn/backends/RefLayerSupport.cpp @@ -135,8 +135,12 @@ bool IsSubtractionSupportedRef(const TensorInfo& input0, const TensorInfo& output, std::string* reasonIfUnsupported) { - // At the moment subtraction is not supported - return false; + ignore_unused(input1); + ignore_unused(output); + return IsSupportedForDataTypeRef(reasonIfUnsupported, + input0.GetDataType(), + &TrueFunc<>, + &TrueFunc<>); } bool IsFullyConnectedSupportedRef(const TensorInfo& input, |