1 files changed, 9 insertions, 3 deletions
diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs
index 753c2443a4..32a9bba5ab 100644
--- a/src/armnnSerializer/ArmnnSchema.fbs
+++ b/src/armnnSerializer/ArmnnSchema.fbs
@@ -171,7 +171,8 @@ enum LayerType : uint {
     Rank = 58,
     LogicalBinary = 59,
     Reduce = 60,
-    Cast = 61
+    Cast = 61,
+    Shape = 62
 }
 
 // Base layer table to be used as part of other layers
@@ -487,7 +488,7 @@ table ReshapeLayer {
 }
 
 table ReshapeDescriptor {
-  targetShape:[uint];
+    targetShape:[uint];
 }
 
 table PermuteLayer {
@@ -499,6 +500,10 @@ table PermuteDescriptor {
     dimMappings:[uint];
 }
 
+table ShapeLayer {
+    base:LayerBase;
+}
+
 table SpaceToBatchNdLayer {
     base:LayerBase;
     descriptor:SpaceToBatchNdDescriptor;
@@ -972,7 +977,8 @@ union Layer {
     RankLayer,
     LogicalBinaryLayer,
     ReduceLayer,
-    CastLayer
+    CastLayer,
+    ShapeLayer
 }
 
 table AnyLayer {