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 a3fdcf8123..39df0c2a7f 100644
--- a/src/armnnSerializer/Serializer.cpp
+++ b/src/armnnSerializer/Serializer.cpp
@@ -48,6 +48,8 @@ serializer::ActivationFunction GetFlatBufferActivationFunction(armnn::Activation
return serializer::ActivationFunction::ActivationFunction_Square;
case armnn::ActivationFunction::Elu:
return serializer::ActivationFunction::ActivationFunction_Elu;
+ case armnn::ActivationFunction::HardSwish:
+ return serializer::ActivationFunction::ActivationFunction_HardSwish;
default:
return serializer::ActivationFunction::ActivationFunction_Sigmoid;
}