IVGCVSW-2702 Add Serializer and Deserializer for Permute
Change-Id: I56322cb935de42cfa25bd1fbb0c09d00c7a5dd2b
Signed-off-by: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
diff --git a/src/armnnSerializer/Schema.fbs b/src/armnnSerializer/Schema.fbs
index e813651..94ca23b 100644
--- a/src/armnnSerializer/Schema.fbs
+++ b/src/armnnSerializer/Schema.fbs
@@ -90,7 +90,8 @@
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 @@
targetShape:[uint];
}
+table PermuteLayer {
+ base:LayerBase;
+ descriptor:PermuteDescriptor;
+}
+
+table PermuteDescriptor {
+ dimMappings:[uint];
+}
+
union Layer {
ActivationLayer,
AdditionLayer,
@@ -233,6 +243,7 @@
InputLayer,
MultiplicationLayer,
OutputLayer,
+ PermuteLayer,
Pooling2dLayer,
ReshapeLayer,
SoftmaxLayer