From 8e7fa232b4e637cc02f2ca344b2113c63cdc7e5a Mon Sep 17 00:00:00 2001
From: ruoyan01 <ruomei.yan@arm.com>
Date: Thu, 28 Feb 2019 15:09:07 +0000
Subject: IVGCVSW-2681 Serialize/de-serialize the BatchNormalization layer

Change-Id: I418c4465366742262fb6e6c1eeba76c634beaeb5
Signed-off-by: ruoyan01 <ruomei.yan@arm.com>
---
 src/armnnSerializer/ArmnnSchema.fbs | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

(limited to 'src/armnnSerializer/ArmnnSchema.fbs')

diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs
index 4e3180f0ff..f416912618 100644
--- a/src/armnnSerializer/ArmnnSchema.fbs
+++ b/src/armnnSerializer/ArmnnSchema.fbs
@@ -103,7 +103,8 @@ enum LayerType : uint {
     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 @@ table PadDescriptor {
     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,
-- 
cgit v1.2.1