From ac97c8cda28f81ce76834b8b769967d42b02e2ac Mon Sep 17 00:00:00 2001
From: Sadik Armagan <sadik.armagan@arm.com>
Date: Mon, 4 Mar 2019 17:44:21 +0000
Subject: IVGCVSW-2696 Serialize / de-serialize the Mean layer

Change-Id: Iee4bab5a6d6b992cf4bba8697a2918f854c906a3
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
---
 src/armnnSerializer/ArmnnSchema.fbs | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

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

diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs
index ac32e661c0..01142ff391 100644
--- a/src/armnnSerializer/ArmnnSchema.fbs
+++ b/src/armnnSerializer/ArmnnSchema.fbs
@@ -109,7 +109,8 @@ enum LayerType : uint {
     ResizeBilinear = 25,
     Subtraction = 26,
     StridedSlice = 27,
-    Gather = 28
+    Gather = 28,
+    Mean = 29
 }
 
 // Base layer table to be used as part of other layers
@@ -349,6 +350,16 @@ table NormalizationDescriptor {
     dataLayout:DataLayout = NCHW;
 }
 
+table MeanLayer {
+    base:LayerBase;
+    descriptor:MeanDescriptor;
+}
+
+table MeanDescriptor {
+    axis:[uint];
+    keepDims:bool = false;
+}
+
 table PadLayer {
     base:LayerBase;
     descriptor:PadDescriptor;
@@ -433,7 +444,8 @@ union Layer {
     ResizeBilinearLayer,
     SubtractionLayer,
     StridedSliceLayer,
-    GatherLayer
+    GatherLayer,
+    MeanLayer
 }
 
 table AnyLayer {
-- 
cgit v1.2.1