diff options
Diffstat (limited to 'src/armnnSerializer/ArmnnSchema.fbs')
-rw-r--r-- | src/armnnSerializer/ArmnnSchema.fbs | 28 |
1 files changed, 25 insertions, 3 deletions
diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs index 1f71ce19f2..aa539b188f 100644 --- a/src/armnnSerializer/ArmnnSchema.fbs +++ b/src/armnnSerializer/ArmnnSchema.fbs @@ -47,6 +47,13 @@ enum DataLayout : byte { NCHW = 1 } +enum ReduceOperation: byte { + Sum = 0, + Max = 1, + Mean = 2, + Min = 3 +} + enum ResizeMethod: byte { NearestNeighbor = 0, Bilinear = 1, @@ -160,7 +167,8 @@ enum LayerType : uint { QLstm = 56, Fill = 57, Rank = 58, - LogicalBinary = 59 + LogicalBinary = 59, + Reduce = 60 } // Base layer table to be used as part of other layers @@ -881,6 +889,19 @@ table RankLayer { base:LayerBase; } +table ReduceLayer { + base:LayerBase; + descriptor:ReduceDescriptor; +} + +table ReduceDescriptor { + targetHeight:uint; + targetWidth:uint; + keepDims:bool = false; + axis:[uint]; + reduceOperation:ReduceOperation = Sum; +} + union Layer { ActivationLayer, AdditionLayer, @@ -941,7 +962,8 @@ union Layer { QLstmLayer, FillLayer, RankLayer, - LogicalBinaryLayer + LogicalBinaryLayer, + ReduceLayer } table AnyLayer { @@ -960,4 +982,4 @@ table SerializedGraph { featureVersions:FeatureCompatibilityVersions; } -root_type SerializedGraph;
\ No newline at end of file +root_type SerializedGraph; |