diff options
author | Matthew Jackson <matthew.jackson@arm.com> | 2019-07-11 15:54:20 +0100 |
---|---|---|
committer | Matthew Jackson <matthew.jackson@arm.com> | 2019-07-16 09:05:14 +0000 |
commit | b5433ee34fd9d38c1453dc062b36348d65677002 (patch) | |
tree | f48ed9f6ffb25aed908c5aa775b49bef78fd906f /src/armnnSerializer/ArmnnSchema.fbs | |
parent | 15a9a8f8f5e05c9967c3a52ecbfb7e173e9e61dd (diff) | |
download | armnn-b5433ee34fd9d38c1453dc062b36348d65677002.tar.gz |
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
Diffstat (limited to 'src/armnnSerializer/ArmnnSchema.fbs')
-rw-r--r-- | src/armnnSerializer/ArmnnSchema.fbs | 17 |
1 files changed, 15 insertions, 2 deletions
diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs index 09187927ae..e3953df83e 100644 --- a/src/armnnSerializer/ArmnnSchema.fbs +++ b/src/armnnSerializer/ArmnnSchema.fbs @@ -130,7 +130,8 @@ enum LayerType : uint { 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 @@ table ResizeDescriptor { dataLayout:DataLayout; } +table StackLayer { + base:LayerBase; + descriptor:StackDescriptor; +} + +table StackDescriptor { + axis:uint; + numInputs:uint; + inputShape:[uint]; +} + union Layer { ActivationLayer, AdditionLayer, @@ -643,7 +655,8 @@ union Layer { SpaceToDepthLayer, PreluLayer, TransposeConvolution2dLayer, - ResizeLayer + ResizeLayer, + StackLayer } table AnyLayer { |