From f195f03e095a5c4dc6880be11af64cab83b5c94b Mon Sep 17 00:00:00 2001 From: David Beck Date: Thu, 6 Sep 2018 16:46:34 +0100 Subject: IVGCVSW-1803 : add Ref Subtraction layer Change-Id: I4c019d626f9369245eca6d549bbe7a28e141f198 --- src/armnn/backends/RefLayerSupport.cpp | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) (limited to 'src/armnn/backends/RefLayerSupport.cpp') 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, -- cgit v1.2.1