commit 452869973b9a45c9c44820d16f92f7dfc96e9aef
author Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> Tue Feb 26 15:53:02 2019 +0000
committer derek.lamberti <derek.lamberti@arm.com> Tue Feb 26 16:49:44 2019 +0000
tree fe4c176852db25435da8865008c626cbd064d7bf
parent 07d58c77e174a8e041c1facc61637e85c6cc0bf6

IVGCVSW-2708 Add Serializer and Deserializer SpaceToBatchNd

Change-Id: I14f2a660654dc87939fe413fafa9384f09594f68
Signed-off-by: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>

src/armnnSerializer/ArmnnSchema.fbs

diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs
index db70e7b..c1e4e38 100644
--- a/src/armnnSerializer/ArmnnSchema.fbs
+++ b/src/armnnSerializer/ArmnnSchema.fbs
@@ -93,7 +93,8 @@
     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 @@
     dimMappings:[uint];
 }
 
+table SpaceToBatchNdLayer {
+    base:LayerBase;
+    descriptor:SpaceToBatchNdDescriptor;
+}
+
+table SpaceToBatchNdDescriptor {
+    blockShape:[uint];
+    padList:[uint];
+    dataLayout:DataLayout;
+}
+
 union Layer {
     ActivationLayer,
     AdditionLayer,
@@ -267,7 +279,8 @@
     PermuteLayer,
     Pooling2dLayer,
     ReshapeLayer,
-    SoftmaxLayer
+    SoftmaxLayer,
+    SpaceToBatchNdLayer
 }
 
 table AnyLayer {