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 048181a2b1..2b96ad8de1 100644
--- a/src/armnnSerializer/Schema.fbs
+++ b/src/armnnSerializer/Schema.fbs
@@ -73,7 +73,8 @@ enum LayerType : uint {
Multiplication = 2,
Output = 3,
Pooling2d = 4,
- Softmax = 5
+ Reshape = 5,
+ Softmax = 6
}
// Base layer table to be used as part of other layers
@@ -152,12 +153,22 @@ table OutputLayer {
base:BindableLayerBase;
}
+table ReshapeLayer {
+ base:LayerBase;
+ descriptor:ReshapeDescriptor;
+}
+
+table ReshapeDescriptor {
+ targetShape:[uint];
+}
+
union Layer {
AdditionLayer,
InputLayer,
MultiplicationLayer,
OutputLayer,
Pooling2dLayer,
+ ReshapeLayer,
SoftmaxLayer
}