aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer/Serializer.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnnSerializer/Serializer.cpp')
-rw-r--r--src/armnnSerializer/Serializer.cpp2
1 files changed, 2 insertions, 0 deletions
diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp
index 6cadb598a2..0df675d1db 100644
--- a/src/armnnSerializer/Serializer.cpp
+++ b/src/armnnSerializer/Serializer.cpp
@@ -78,6 +78,8 @@ serializer::ActivationFunction GetFlatBufferActivationFunction(armnn::Activation
return serializer::ActivationFunction::ActivationFunction_Elu;
case armnn::ActivationFunction::HardSwish:
return serializer::ActivationFunction::ActivationFunction_HardSwish;
+ case armnn::ActivationFunction::Gelu:
+ return serializer::ActivationFunction::ActivationFunction_Gelu;
default:
return serializer::ActivationFunction::ActivationFunction_Sigmoid;
}