diff options
Diffstat (limited to 'src/armnnSerializer/ArmnnSchema.fbs')
-rw-r--r-- | src/armnnSerializer/ArmnnSchema.fbs | 26 |
1 files changed, 25 insertions, 1 deletions
diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs index 0fd8da7e8f..513c74e82d 100644 --- a/src/armnnSerializer/ArmnnSchema.fbs +++ b/src/armnnSerializer/ArmnnSchema.fbs @@ -131,7 +131,8 @@ enum LayerType : uint { Prelu = 41, TransposeConvolution2d = 42, Resize = 43, - Stack = 44 + Stack = 44, + QuantizedLstm = 45 } // Base layer table to be used as part of other layers @@ -544,6 +545,23 @@ table LstmInputParams { outputLayerNormWeights:ConstTensor; } +table QuantizedLstmInputParams { + inputToInputWeights:ConstTensor; + inputToForgetWeights:ConstTensor; + inputToCellWeights:ConstTensor; + inputToOutputWeights:ConstTensor; + + recurrentToInputWeights:ConstTensor; + recurrentToForgetWeights:ConstTensor; + recurrentToCellWeights:ConstTensor; + recurrentToOutputWeights:ConstTensor; + + inputGateBias:ConstTensor; + forgetGateBias:ConstTensor; + cellBias:ConstTensor; + outputGateBias:ConstTensor; +} + table LstmDescriptor { activationFunc:uint; clippingThresCell:float; @@ -560,6 +578,11 @@ table LstmLayer { inputParams:LstmInputParams; } +table QuantizedLstmLayer { + base:LayerBase; + inputParams:QuantizedLstmInputParams; +} + table DequantizeLayer { base:LayerBase; } @@ -653,6 +676,7 @@ union Layer { SplitterLayer, DetectionPostProcessLayer, LstmLayer, + QuantizedLstmLayer, QuantizeLayer, DequantizeLayer, MergeLayer, |