IVGCVSW-2700 Serialize/de-serialize the Normalization layer

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

1 files changed, 29 insertions, 2 deletions

diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs
index b59adcf82b..cde0087d6f 100644
--- a/src/armnnSerializer/ArmnnSchema.fbs
+++ b/src/armnnSerializer/ArmnnSchema.fbs
@@ -99,7 +99,8 @@ enum LayerType : uint {
     Division = 15,
     Minimum = 16,
     Equal = 17,
-    Maximum = 18
+    Maximum = 18,
+    Normalization = 19
 }
 
 // Base layer table to be used as part of other layers
@@ -298,6 +299,31 @@ table BatchToSpaceNdDescriptor {
     dataLayout:DataLayout;
 }
 
+enum NormalizationAlgorithmChannel : byte {
+    Across = 0,
+    Within = 1
+}
+
+enum NormalizationAlgorithmMethod : byte {
+    LocalBrightness = 0,
+    LocalContrast = 1
+}
+
+table NormalizationLayer {
+    base:LayerBase;
+    descriptor:NormalizationDescriptor;
+}
+
+table NormalizationDescriptor {
+    normChannelType:NormalizationAlgorithmChannel = Across;
+    normMethodType:NormalizationAlgorithmMethod = LocalBrightness;
+    normSize:uint;
+    alpha:float;
+    beta:float;
+    k:float;
+    dataLayout:DataLayout = NCHW;
+}
+
 union Layer {
     ActivationLayer,
     AdditionLayer,
@@ -317,7 +343,8 @@ union Layer {
     DivisionLayer,
     MinimumLayer,
     EqualLayer,
-    MaximumLayer
+    MaximumLayer,
+    NormalizationLayer
 }
 
 table AnyLayer {