commit 57728788f65656e4fa08923d12bee0de34a72fc7
author Nina Drozd <nina.drozd@arm.com> Wed Feb 27 10:53:27 2019 +0000
committer Nina Drozd <nina.drozd@arm.com> Fri Mar 01 10:52:26 2019 +0000
tree 4d84f00762483cb26f6c2e880a07eb8b7699b7da
parent 377351e5420304668da92da4ee00a012923619d1

IVGCVSW-2700 Serialize/de-serialize the Normalization layer

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

src/armnnSerializer/ArmnnSchema.fbs

diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs
index b59adcf..cde0087 100644
--- a/src/armnnSerializer/ArmnnSchema.fbs
+++ b/src/armnnSerializer/ArmnnSchema.fbs
@@ -99,7 +99,8 @@
     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 @@
     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 @@
     DivisionLayer,
     MinimumLayer,
     EqualLayer,
-    MaximumLayer
+    MaximumLayer,
+    NormalizationLayer
 }
 
 table AnyLayer {