IVGCVSW-2700 Serialize/de-serialize the Normalization layer

Change-Id: Ib307ec6c28beb6c158d337678e67a2484c495a06
Signed-off-by: Nina Drozd <nina.drozd@arm.com>

1 files changed, 28 insertions, 0 deletions

diff --git a/src/armnnSerializer/SerializerUtils.cpp b/src/armnnSerializer/SerializerUtils.cpp
index 592f258b81..bfe795c8c4 100644
--- a/src/armnnSerializer/SerializerUtils.cpp
+++ b/src/armnnSerializer/SerializerUtils.cpp
@@ -96,4 +96,32 @@ armnnSerializer::PaddingMethod GetFlatBufferPaddingMethod(armnn::PaddingMethod p
     }
 }
 
+armnnSerializer::NormalizationAlgorithmChannel GetFlatBufferNormalizationAlgorithmChannel(
+    armnn::NormalizationAlgorithmChannel normalizationAlgorithmChannel)
+{
+    switch (normalizationAlgorithmChannel)
+    {
+        case armnn::NormalizationAlgorithmChannel::Across:
+            return armnnSerializer::NormalizationAlgorithmChannel::NormalizationAlgorithmChannel_Across;
+        case armnn::NormalizationAlgorithmChannel::Within:
+            return armnnSerializer::NormalizationAlgorithmChannel::NormalizationAlgorithmChannel_Within;
+        default:
+            return armnnSerializer::NormalizationAlgorithmChannel::NormalizationAlgorithmChannel_Across;
+    }
+}
+
+armnnSerializer::NormalizationAlgorithmMethod GetFlatBufferNormalizationAlgorithmMethod(
+    armnn::NormalizationAlgorithmMethod normalizationAlgorithmMethod)
+{
+    switch (normalizationAlgorithmMethod)
+    {
+        case armnn::NormalizationAlgorithmMethod::LocalBrightness:
+            return armnnSerializer::NormalizationAlgorithmMethod::NormalizationAlgorithmMethod_LocalBrightness;
+        case armnn::NormalizationAlgorithmMethod::LocalContrast:
+            return armnnSerializer::NormalizationAlgorithmMethod::NormalizationAlgorithmMethod_LocalContrast;
+        default:
+            return armnnSerializer::NormalizationAlgorithmMethod::NormalizationAlgorithmMethod_LocalBrightness;
+    }
+}
+
 } // namespace armnnSerializer
\ No newline at end of file