aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer/Schema.fbs
diff options
context:
space:
mode:
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