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 e81365166f..94ca23b0cd 100644 --- a/src/armnnSerializer/Schema.fbs +++ b/src/armnnSerializer/Schema.fbs @@ -90,7 +90,8 @@ enum LayerType : uint { Softmax = 6, Convolution2d = 7, DepthwiseConvolution2d = 8, - Activation = 9 + Activation = 9, + Permute = 10 } // Base layer table to be used as part of other layers @@ -225,6 +226,15 @@ table ReshapeDescriptor { targetShape:[uint]; } +table PermuteLayer { + base:LayerBase; + descriptor:PermuteDescriptor; +} + +table PermuteDescriptor { + dimMappings:[uint]; +} + union Layer { ActivationLayer, AdditionLayer, @@ -233,6 +243,7 @@ union Layer { InputLayer, MultiplicationLayer, OutputLayer, + PermuteLayer, Pooling2dLayer, ReshapeLayer, SoftmaxLayer |