IVGCVSW-3364 Add serialization support for Resize layer

Signed-off-by: FinnWilliamsArm <Finn.Williams@arm.com>
Change-Id: I3b1af816cefc1760f63324f365de93f899c9c9ee

1 files changed, 21 insertions, 2 deletions

diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs
index 7969d10598..09187927ae 100644
--- a/src/armnnSerializer/ArmnnSchema.fbs
+++ b/src/armnnSerializer/ArmnnSchema.fbs
@@ -36,6 +36,11 @@ enum DataLayout : byte {
     NCHW = 1
 }
 
+enum ResizeMethod: byte {
+    NearestNeighbor = 0,
+    Bilinear = 1,
+}
+
 table TensorInfo {
     dimensions:[uint];
     dataType:DataType;
@@ -124,7 +129,8 @@ enum LayerType : uint {
     Concat = 39,
     SpaceToDepth = 40,
     Prelu = 41,
-    TransposeConvolution2d = 42
+    TransposeConvolution2d = 42,
+    Resize = 43
 }
 
 // Base layer table to be used as part of other layers
@@ -581,6 +587,18 @@ table TransposeConvolution2dDescriptor {
     dataLayout:DataLayout = NCHW;
 }
 
+table ResizeLayer {
+    base:LayerBase;
+    descriptor:ResizeDescriptor;
+}
+
+table ResizeDescriptor {
+    targetHeight:uint;
+    targetWidth:uint;
+    method:ResizeMethod = NearestNeighbor;
+    dataLayout:DataLayout;
+}
+
 union Layer {
     ActivationLayer,
     AdditionLayer,
@@ -624,7 +642,8 @@ union Layer {
     ConcatLayer,
     SpaceToDepthLayer,
     PreluLayer,
-    TransposeConvolution2dLayer
+    TransposeConvolution2dLayer,
+    ResizeLayer
 }
 
 table AnyLayer {