diff options
Diffstat (limited to 'src/armnn/LayerSupport.cpp')
| -rw-r--r-- | src/armnn/LayerSupport.cpp | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/src/armnn/LayerSupport.cpp b/src/armnn/LayerSupport.cpp index dac88385b1..08d91fc20b 100644 --- a/src/armnn/LayerSupport.cpp +++ b/src/armnn/LayerSupport.cpp @@ -570,7 +570,11 @@ bool IsRsqrtSupported(const BackendId& backend, char* reasonIfUnsupported, size_t reasonIfUnsupportedMaxLength) { - FORWARD_LAYER_SUPPORT_FUNC(backend, IsRsqrtSupported, input, output); + FORWARD_LAYER_SUPPORT_FUNC(backend, + IsElementwiseUnarySupported, + input, + output, + ElementwiseUnaryDescriptor(UnaryOperation::Rsqrt)); } bool IsSoftmaxSupported(const BackendId& backend, |