1 files changed, 14 insertions, 2 deletions
diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs
index 5a001de545..1204ad8f0f 100644
--- a/src/armnnSerializer/ArmnnSchema.fbs
+++ b/src/armnnSerializer/ArmnnSchema.fbs
@@ -121,7 +121,8 @@ enum LayerType : uint {
     Dequantize = 36,
     Merge = 37,
     Switch = 38,
-    Concat = 39
+    Concat = 39,
+    SpaceToDepth = 40
 }
 
 // Base layer table to be used as part of other layers
@@ -338,6 +339,16 @@ table SpaceToBatchNdDescriptor {
     dataLayout:DataLayout;
 }
 
+table SpaceToDepthLayer {
+    base:LayerBase;
+    descriptor:SpaceToDepthDescriptor;
+}
+
+table SpaceToDepthDescriptor {
+    blockSize:uint;
+    dataLayout:DataLayout;
+}
+
 table SubtractionLayer {
     base:LayerBase;
 }
@@ -584,7 +595,8 @@ union Layer {
     DequantizeLayer,
     MergeLayer,
     SwitchLayer,
-    ConcatLayer
+    ConcatLayer,
+    SpaceToDepthLayer
 }
 
 table AnyLayer {