diff options
author | David Beck <david.beck@arm.com> | 2018-09-06 16:46:34 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-09-25 14:54:29 +0100 |
commit | f195f03e095a5c4dc6880be11af64cab83b5c94b (patch) | |
tree | d1c6d7d46ed70b915772bd50c5074d13443d9bca /src/armnn/backends/RefLayerSupport.cpp | |
parent | c2044fe9d26a8b6afca48aee04bd5d29f8e27b8d (diff) | |
download | armnn-f195f03e095a5c4dc6880be11af64cab83b5c94b.tar.gz |
IVGCVSW-1803 : add Ref Subtraction layer
Change-Id: I4c019d626f9369245eca6d549bbe7a28e141f198
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, |