aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer/ArmnnSchema.fbs
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnnSerializer/ArmnnSchema.fbs')
-rw-r--r--src/armnnSerializer/ArmnnSchema.fbs16
1 files changed, 14 insertions, 2 deletions
diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs
index ac32e661c0..01142ff391 100644
--- a/src/armnnSerializer/ArmnnSchema.fbs
+++ b/src/armnnSerializer/ArmnnSchema.fbs
@@ -109,7 +109,8 @@ enum LayerType : uint {
ResizeBilinear = 25,
Subtraction = 26,
StridedSlice = 27,
- Gather = 28
+ Gather = 28,
+ Mean = 29
}
// Base layer table to be used as part of other layers
@@ -349,6 +350,16 @@ table NormalizationDescriptor {
dataLayout:DataLayout = NCHW;
}
+table MeanLayer {
+ base:LayerBase;
+ descriptor:MeanDescriptor;
+}
+
+table MeanDescriptor {
+ axis:[uint];
+ keepDims:bool = false;
+}
+
table PadLayer {
base:LayerBase;
descriptor:PadDescriptor;
@@ -433,7 +444,8 @@ union Layer {
ResizeBilinearLayer,
SubtractionLayer,
StridedSliceLayer,
- GatherLayer
+ GatherLayer,
+ MeanLayer
}
table AnyLayer {