aboutsummaryrefslogtreecommitdiff
path: root/delegate/opaque/src/Activation.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'delegate/opaque/src/Activation.hpp')
-rw-r--r--delegate/opaque/src/Activation.hpp9
1 files changed, 9 insertions, 0 deletions
diff --git a/delegate/opaque/src/Activation.hpp b/delegate/opaque/src/Activation.hpp
index 9fce7a12e0..f56609001a 100644
--- a/delegate/opaque/src/Activation.hpp
+++ b/delegate/opaque/src/Activation.hpp
@@ -166,6 +166,15 @@ TfLiteStatus VisitActivationOperator(DelegateData& delegateData,
activationDesc.m_Function = armnn::ActivationFunction::HardSwish;
break;
}
+ case kTfLiteBuiltinLeakyRelu:
+ {
+ // Get alpha param from builtin data
+ auto* leakyReluParameters =
+ reinterpret_cast<TfLiteLeakyReluParams*>(TfLiteOpaqueNodeGetBuiltinData(tfLiteNode));
+ activationDesc.m_Function = armnn::ActivationFunction::LeakyReLu;
+ activationDesc.m_A = leakyReluParameters->alpha;
+ break;
+ }
default:
{
return kTfLiteError;