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