commit b5433ee34fd9d38c1453dc062b36348d65677002
author Matthew Jackson <matthew.jackson@arm.com> Thu Jul 11 15:54:20 2019 +0100
committer Matthew Jackson <matthew.jackson@arm.com> Tue Jul 16 09:05:14 2019 +0000
tree f48ed9f6ffb25aed908c5aa775b49bef78fd906f
parent 15a9a8f8f5e05c9967c3a52ecbfb7e173e9e61dd

IVGCVSW-3420 Add Serialization support for the new Stack layer

 * Adds serialization/deserialization support
 * Adds related unit test

Signed-off-by: Matthew Jackson <matthew.jackson@arm.com>
Change-Id: I69deb5397b8a06c679715e24971e9bb1c282140d

src/armnnSerializer/ArmnnSchema.fbs

diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs
index 0918792..e3953df 100644
--- a/src/armnnSerializer/ArmnnSchema.fbs
+++ b/src/armnnSerializer/ArmnnSchema.fbs
@@ -130,7 +130,8 @@
     SpaceToDepth = 40,
     Prelu = 41,
     TransposeConvolution2d = 42,
-    Resize = 43
+    Resize = 43,
+    Stack = 44
 }
 
 // Base layer table to be used as part of other layers
@@ -599,6 +600,17 @@
     dataLayout:DataLayout;
 }
 
+table StackLayer {
+    base:LayerBase;
+    descriptor:StackDescriptor;
+}
+
+table StackDescriptor {
+    axis:uint;
+    numInputs:uint;
+    inputShape:[uint];
+}
+
 union Layer {
     ActivationLayer,
     AdditionLayer,
@@ -643,7 +655,8 @@
     SpaceToDepthLayer,
     PreluLayer,
     TransposeConvolution2dLayer,
-    ResizeLayer
+    ResizeLayer,
+    StackLayer
 }
 
 table AnyLayer {