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 b43f26c652..3c01842c95 100644
--- a/src/armnnSerializer/Serializer.cpp
+++ b/src/armnnSerializer/Serializer.cpp
@@ -46,6 +46,8 @@ serializer::ActivationFunction GetFlatBufferActivationFunction(armnn::Activation
return serializer::ActivationFunction::ActivationFunction_Sqrt;
case armnn::ActivationFunction::Square:
return serializer::ActivationFunction::ActivationFunction_Square;
+ case armnn::ActivationFunction::Elu:
+ return serializer::ActivationFunction::ActivationFunction_Elu;
default:
return serializer::ActivationFunction::ActivationFunction_Sigmoid;
}