diff options
Diffstat (limited to 'src/armnnSerializer/ArmnnSchema.fbs')
-rw-r--r-- | src/armnnSerializer/ArmnnSchema.fbs | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs index 6a388db699..e1b6e1f768 100644 --- a/src/armnnSerializer/ArmnnSchema.fbs +++ b/src/armnnSerializer/ArmnnSchema.fbs @@ -59,6 +59,7 @@ table TensorInfo { quantizationOffset:int = 0; quantizationScales:[float]; quantizationDim:uint; + dimensionality:uint = 1; } struct Connection { @@ -157,7 +158,8 @@ enum LayerType : uint { ElementwiseUnary = 54, Transpose = 55, QLstm = 56, - Fill = 57 + Fill = 57, + Rank = 58 } // Base layer table to be used as part of other layers @@ -859,6 +861,10 @@ table StandInLayer { descriptor:StandInDescriptor; } +table RankLayer { + base:LayerBase; +} + union Layer { ActivationLayer, AdditionLayer, @@ -917,7 +923,8 @@ union Layer { ElementwiseUnaryLayer, TransposeLayer, QLstmLayer, - FillLayer + FillLayer, + RankLayer } table AnyLayer { |