diff options
author | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-02-20 17:31:34 +0000 |
---|---|---|
committer | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-02-20 17:31:34 +0000 |
commit | 30b0020478652e441a5dff4880261f7c7007bb6c (patch) | |
tree | fed46af6b7a62acdcd9ffb906e00efb679572883 /src/armnnSerializer/Schema.fbs | |
parent | af484013329a8ca5b3c4c9d16395fb79dd19b1b2 (diff) | |
download | armnn-30b0020478652e441a5dff4880261f7c7007bb6c.tar.gz |
IVGCVSW-2702 Add Serializer and Deserializer for Permute
Change-Id: I56322cb935de42cfa25bd1fbb0c09d00c7a5dd2b
Signed-off-by: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
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 |