diff options
Diffstat (limited to 'src/armnnSerializer/Schema.fbs')
-rw-r--r-- | src/armnnSerializer/Schema.fbs | 13 |
1 files changed, 12 insertions, 1 deletions
diff --git a/src/armnnSerializer/Schema.fbs b/src/armnnSerializer/Schema.fbs index 048181a2b1..2b96ad8de1 100644 --- a/src/armnnSerializer/Schema.fbs +++ b/src/armnnSerializer/Schema.fbs @@ -73,7 +73,8 @@ enum LayerType : uint { Multiplication = 2, Output = 3, Pooling2d = 4, - Softmax = 5 + Reshape = 5, + Softmax = 6 } // Base layer table to be used as part of other layers @@ -152,12 +153,22 @@ table OutputLayer { base:BindableLayerBase; } +table ReshapeLayer { + base:LayerBase; + descriptor:ReshapeDescriptor; +} + +table ReshapeDescriptor { + targetShape:[uint]; +} + union Layer { AdditionLayer, InputLayer, MultiplicationLayer, OutputLayer, Pooling2dLayer, + ReshapeLayer, SoftmaxLayer } |