IVGCVSW-2642 Add Reshape to Serializer and Deserializer

Change-Id: Iccded3c6e3c0713c44f43231981440420591f94e
Signed-off-by: Saoirse Stewart <saoirse.stewart@arm.com>
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>

1 files changed, 12 insertions, 1 deletions

diff --git a/src/armnnSerializer/Schema.fbs b/src/armnnSerializer/Schema.fbs
index 048181a2b1..2b96ad8de1 100644
--- a/src/armnnSerializer/Schema.fbs
+++ b/src/armnnSerializer/Schema.fbs
@@ -73,7 +73,8 @@ enum LayerType : uint {
     Multiplication = 2,
     Output = 3,
     Pooling2d = 4,
-    Softmax = 5
+    Reshape = 5,
+    Softmax = 6
 }
 
 // Base layer table to be used as part of other layers
@@ -152,12 +153,22 @@ table OutputLayer {
     base:BindableLayerBase;
 }
 
+table ReshapeLayer {
+    base:LayerBase;
+    descriptor:ReshapeDescriptor;
+}
+
+table ReshapeDescriptor {
+  targetShape:[uint];
+}
+
 union Layer {
     AdditionLayer,
     InputLayer,
     MultiplicationLayer,
     OutputLayer,
     Pooling2dLayer,
+    ReshapeLayer,
     SoftmaxLayer
 }