commit cb549301bc4c5a405e02c1f433537557423d2e36
author Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> Fri Jun 21 13:53:38 2019 +0100
committer Áron Virginás-Tar <aron.virginas-tar@arm.com> Fri Jun 21 14:30:08 2019 +0000
tree 8f9bf05911c05e673e0b90b68c29164e47e8609c
parent 0dcffec80292cd2e0e7c2736fd3db63abd7c3f64

IVGCVSW-3321 Add serialization support for TransposeConvolution2dLayer

Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: If0c8f3662d5e03696f97040abed784c0fbcdbc6f

src/armnnSerializer/ArmnnSchema.fbs

diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs
index 7947893..83275ca 100644
--- a/src/armnnSerializer/ArmnnSchema.fbs
+++ b/src/armnnSerializer/ArmnnSchema.fbs
@@ -123,7 +123,8 @@
     Switch = 38,
     Concat = 39,
     SpaceToDepth = 40,
-    Prelu = 41
+    Prelu = 41,
+    TransposeConvolution2d = 42
 }
 
 // Base layer table to be used as part of other layers
@@ -561,6 +562,24 @@
     base:LayerBase;
 }
 
+table TransposeConvolution2dLayer {
+    base:LayerBase;
+    descriptor:TransposeConvolution2dDescriptor;
+    weights:ConstTensor;
+    biases:ConstTensor;
+}
+
+table TransposeConvolution2dDescriptor {
+    padLeft:uint;
+    padRight:uint;
+    padTop:uint;
+    padBottom:uint;
+    strideX:uint;
+    strideY:uint;
+    biasEnabled:bool = false;
+    dataLayout:DataLayout = NCHW;
+}
+
 union Layer {
     ActivationLayer,
     AdditionLayer,
@@ -603,7 +622,8 @@
     SwitchLayer,
     ConcatLayer,
     SpaceToDepthLayer,
-    PreluLayer
+    PreluLayer,
+    TransposeConvolution2dLayer
 }
 
 table AnyLayer {