IVGCVSW-2710 Add Serializer and Deserializer for StridedSlice
Change-Id: I6e8198a2aa5f8c56f00ccf9b4d98fcd208755654
Signed-off-by: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs
index 9f1d8ec..ed3de83 100644
--- a/src/armnnSerializer/ArmnnSchema.fbs
+++ b/src/armnnSerializer/ArmnnSchema.fbs
@@ -107,7 +107,8 @@
BatchNormalization = 23,
Greater = 24,
ResizeBilinear = 25,
- Subtraction = 26
+ Subtraction = 26,
+ StridedSlice = 27
}
// Base layer table to be used as part of other layers
@@ -381,6 +382,23 @@
dataLayout:DataLayout;
}
+table StridedSliceLayer {
+ base:LayerBase;
+ descriptor:StridedSliceDescriptor;
+}
+
+table StridedSliceDescriptor {
+ begin:[int];
+ end:[int];
+ stride:[int];
+ beginMask:int;
+ endMask:int;
+ shrinkAxisMask:int;
+ ellipsisMask:int;
+ newAxisMask:int;
+ dataLayout:DataLayout;
+}
+
union Layer {
ActivationLayer,
AdditionLayer,
@@ -408,7 +426,8 @@
FloorLayer,
GreaterLayer,
ResizeBilinearLayer,
- SubtractionLayer
+ SubtractionLayer,
+ StridedSliceLayer
}
table AnyLayer {