2 files changed, 4 insertions, 1 deletions
diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs
index d175d41f3f..d7565a5b9a 100644
--- a/src/armnnSerializer/ArmnnSchema.fbs
+++ b/src/armnnSerializer/ArmnnSchema.fbs
@@ -20,7 +20,8 @@ enum ActivationFunction : byte {
     Abs = 7,
     Sqrt = 8,
     Square = 9,
-    Elu = 10
+    Elu = 10,
+    HardSwish = 11
 }
 
 enum ArgMinMaxFunction : byte {
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;
     }