diff options
author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-06-21 13:53:38 +0100 |
---|---|---|
committer | Áron Virginás-Tar <aron.virginas-tar@arm.com> | 2019-06-21 14:30:08 +0000 |
commit | cb549301bc4c5a405e02c1f433537557423d2e36 (patch) | |
tree | 8f9bf05911c05e673e0b90b68c29164e47e8609c /src/armnnSerializer/ArmnnSchema.fbs | |
parent | 0dcffec80292cd2e0e7c2736fd3db63abd7c3f64 (diff) | |
download | armnn-cb549301bc4c5a405e02c1f433537557423d2e36.tar.gz |
IVGCVSW-3321 Add serialization support for TransposeConvolution2dLayer
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: If0c8f3662d5e03696f97040abed784c0fbcdbc6f
Diffstat (limited to 'src/armnnSerializer/ArmnnSchema.fbs')
-rw-r--r-- | src/armnnSerializer/ArmnnSchema.fbs | 24 |
1 files changed, 22 insertions, 2 deletions
diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs index 794789390a..83275ca248 100644 --- a/src/armnnSerializer/ArmnnSchema.fbs +++ b/src/armnnSerializer/ArmnnSchema.fbs @@ -123,7 +123,8 @@ enum LayerType : uint { 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 @@ table PreluLayer { 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 @@ union Layer { SwitchLayer, ConcatLayer, SpaceToDepthLayer, - PreluLayer + PreluLayer, + TransposeConvolution2dLayer } table AnyLayer { |