diff options
Diffstat (limited to 'src/armnn/LayerSupport.cpp')
-rw-r--r-- | src/armnn/LayerSupport.cpp | 14 |
1 files changed, 12 insertions, 2 deletions
diff --git a/src/armnn/LayerSupport.cpp b/src/armnn/LayerSupport.cpp index f88e4e1cc9..997b5f245a 100644 --- a/src/armnn/LayerSupport.cpp +++ b/src/armnn/LayerSupport.cpp @@ -262,7 +262,12 @@ bool IsEqualSupported(const BackendId& backend, char* reasonIfUnsupported, size_t reasonIfUnsupportedMaxLength) { - FORWARD_LAYER_SUPPORT_FUNC(backend, IsEqualSupported, input0, input1, output); + FORWARD_LAYER_SUPPORT_FUNC(backend, + IsComparisonSupported, + input0, + input1, + output, + ComparisonDescriptor(ComparisonOperation::Equal)); } bool IsFakeQuantizationSupported(const BackendId& backend, @@ -317,7 +322,12 @@ bool IsGreaterSupported(const BackendId& backend, char* reasonIfUnsupported, size_t reasonIfUnsupportedMaxLength) { - FORWARD_LAYER_SUPPORT_FUNC(backend, IsGreaterSupported, input0, input1, output); + FORWARD_LAYER_SUPPORT_FUNC(backend, + IsComparisonSupported, + input0, + input1, + output, + ComparisonDescriptor(ComparisonOperation::Greater)); } bool IsInputSupported(const BackendId& backend, |