IVGCVSW-4375 Add support for Transpose
* Added TransposeLayer
* Added CL, Neon and Ref Workloads
* Added Transpose utilities
* Added Serializer and Deserializer support
* Added Quantizer support
Signed-off-by: Mike Kelly <mike.kelly@arm.com>
Change-Id: I04c755ba7cb5b1edf72b3c9f3c0314878032e3c7
diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs
index 0697517..d175d41 100644
--- a/src/armnnSerializer/ArmnnSchema.fbs
+++ b/src/armnnSerializer/ArmnnSchema.fbs
@@ -149,7 +149,8 @@
LogSoftmax = 51,
Comparison = 52,
StandIn = 53,
- ElementwiseUnary = 54
+ ElementwiseUnary = 54,
+ Transpose = 55
}
// Base layer table to be used as part of other layers
@@ -732,6 +733,15 @@
dataLayout:DataLayout = NCHW;
}
+table TransposeLayer {
+ base:LayerBase;
+ descriptor:TransposeDescriptor;
+}
+
+table TransposeDescriptor {
+ dimMappings:[uint];
+}
+
table ResizeLayer {
base:LayerBase;
descriptor:ResizeDescriptor;
@@ -820,7 +830,8 @@
LogSoftmaxLayer,
ComparisonLayer,
StandInLayer,
- ElementwiseUnaryLayer
+ ElementwiseUnaryLayer,
+ TransposeLayer
}
table AnyLayer {