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.hpp5
1 files changed, 5 insertions, 0 deletions
diff --git a/delegate/classic/src/Activation.hpp b/delegate/classic/src/Activation.hpp
index a93cee43a3..1c55c2e9b5 100644
--- a/delegate/classic/src/Activation.hpp
+++ b/delegate/classic/src/Activation.hpp
@@ -109,6 +109,11 @@ TfLiteStatus VisitActivationOperator(DelegateData& delegateData,
activationDesc.m_A = leakyReluParameters->alpha;
break;
}
+ case kTfLiteBuiltinGelu:
+ {
+ activationDesc.m_Function = armnn::ActivationFunction::Gelu;
+ break;
+ }
default:
{
return kTfLiteError;