diff options
author | Jan Eilers <jan.eilers@arm.com> | 2019-07-23 09:47:43 +0100 |
---|---|---|
committer | Jan Eilers <jan.eilers@arm.com> | 2019-07-29 16:19:23 +0000 |
commit | 5b01a8994caea2857f3b991dc69a814f12ab7743 (patch) | |
tree | 434660d1ba049de847ee7b5ff9715bb618421831 /src/armnnSerializer/ArmnnSchema.fbs | |
parent | 61e71aa399a93cec44b23d43f2293e18d00f8e3a (diff) | |
download | armnn-5b01a8994caea2857f3b991dc69a814f12ab7743.tar.gz |
IVGCVSW-3471 Add Serialization support for Quantized_LSTM
* Adds serialization/deserialization support
* Adds related Unit test
Signed-off-by: Jan Eilers <jan.eilers@arm.com>
Change-Id: Iaf271aa7d848bc3a69dbbf182389f2241c0ced5f
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, |