commit 8e7fa232b4e637cc02f2ca344b2113c63cdc7e5a
author ruoyan01 <ruomei.yan@arm.com> Thu Feb 28 15:09:07 2019 +0000
committer ruoyan01 <ruomei.yan@arm.com> Fri Mar 01 16:39:27 2019 +0000
tree 3c200afe3c7cab37b553ba0461aed4410b7cfbb8
parent dd2ba7ebf78a75aadd8ddd2ae1a4226ffc4ae4d9

IVGCVSW-2681 Serialize/de-serialize the BatchNormalization layer

Change-Id: I418c4465366742262fb6e6c1eeba76c634beaeb5
Signed-off-by: ruoyan01 <ruomei.yan@arm.com>

src/armnnSerializer/ArmnnSchema.fbs

diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs
index 4e3180f..f416912 100644
--- a/src/armnnSerializer/ArmnnSchema.fbs
+++ b/src/armnnSerializer/ArmnnSchema.fbs
@@ -103,7 +103,8 @@
     Normalization = 19,
     Pad = 20,
     Rsqrt = 21,
-    Floor = 22
+    Floor = 22,
+    BatchNormalization = 23
 }
 
 // Base layer table to be used as part of other layers
@@ -340,14 +341,30 @@
     padList:[uint];
 }
 
+
 table RsqrtLayer {
     base:LayerBase;
 }
 
+table BatchNormalizationLayer {
+    base:LayerBase;
+    descriptor:BatchNormalizationDescriptor;
+    mean:ConstTensor;
+    variance:ConstTensor;
+    beta:ConstTensor;
+    gamma:ConstTensor;
+}
+
+table BatchNormalizationDescriptor {
+    eps:float;
+    dataLayout:DataLayout;
+}
+
 union Layer {
     ActivationLayer,
     AdditionLayer,
     BatchToSpaceNdLayer,
+    BatchNormalizationLayer,
     ConstantLayer,
     Convolution2dLayer,
     DepthwiseConvolution2dLayer,