From 452869973b9a45c9c44820d16f92f7dfc96e9aef Mon Sep 17 00:00:00 2001
From: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
Date: Tue, 26 Feb 2019 15:53:02 +0000
Subject: IVGCVSW-2708 Add Serializer and Deserializer SpaceToBatchNd

Change-Id: I14f2a660654dc87939fe413fafa9384f09594f68
Signed-off-by: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
---
 src/armnnSerializer/ArmnnSchema.fbs | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

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

diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs
index db70e7bd4c..c1e4e3826a 100644
--- a/src/armnnSerializer/ArmnnSchema.fbs
+++ b/src/armnnSerializer/ArmnnSchema.fbs
@@ -93,7 +93,8 @@ enum LayerType : uint {
     Activation = 9,
     Permute = 10,
     FullyConnected = 11,
-    Constant = 12
+    Constant = 12,
+    SpaceToBatchNd = 13
 }
 
 // Base layer table to be used as part of other layers
@@ -254,6 +255,17 @@ table PermuteDescriptor {
     dimMappings:[uint];
 }
 
+table SpaceToBatchNdLayer {
+    base:LayerBase;
+    descriptor:SpaceToBatchNdDescriptor;
+}
+
+table SpaceToBatchNdDescriptor {
+    blockShape:[uint];
+    padList:[uint];
+    dataLayout:DataLayout;
+}
+
 union Layer {
     ActivationLayer,
     AdditionLayer,
@@ -267,7 +279,8 @@ union Layer {
     PermuteLayer,
     Pooling2dLayer,
     ReshapeLayer,
-    SoftmaxLayer
+    SoftmaxLayer,
+    SpaceToBatchNdLayer
 }
 
 table AnyLayer {
-- 
cgit v1.2.1