aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer/Schema.fbs
diff options
context:
space:
mode:
authorNattapat Chaimanowong <nattapat.chaimanowong@arm.com>2019-02-20 17:31:34 +0000
committerNattapat Chaimanowong <nattapat.chaimanowong@arm.com>2019-02-20 17:31:34 +0000
commit30b0020478652e441a5dff4880261f7c7007bb6c (patch)
treefed46af6b7a62acdcd9ffb906e00efb679572883 /src/armnnSerializer/Schema.fbs
parentaf484013329a8ca5b3c4c9d16395fb79dd19b1b2 (diff)
downloadarmnn-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.fbs13
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