diff options
author | Jan Eilers <jan.eilers@arm.com> | 2019-07-17 11:07:49 +0100 |
---|---|---|
committer | Jan Eilers <jan.eilers@arm.com> | 2019-07-23 11:01:29 +0100 |
commit | f8c629701e760e2476582c91e6b7f5e1313dc02a (patch) | |
tree | 3c9dc1a673fe99367a7f25d98199d8ab386e8bc2 /src/armnnSerializer/ArmnnSchema.fbs | |
parent | 07f2121feeeeae36a7e67eeb8a6965df63b848f3 (diff) | |
download | armnn-f8c629701e760e2476582c91e6b7f5e1313dc02a.tar.gz |
IVGCVSW-3526 Add layer norm support for lstm serialization
* Adds layer norm support for serialization/deserialization
* Adds related unit tests
Change-Id: If80b668accc8b0754a93d18ab3a243284cb383d1
Signed-off-by: Jan Eilers <jan.eilers@arm.com>
Diffstat (limited to 'src/armnnSerializer/ArmnnSchema.fbs')
-rw-r--r-- | src/armnnSerializer/ArmnnSchema.fbs | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs index e3953df83e..0fd8da7e8f 100644 --- a/src/armnnSerializer/ArmnnSchema.fbs +++ b/src/armnnSerializer/ArmnnSchema.fbs @@ -537,6 +537,11 @@ table LstmInputParams { cellToForgetWeights:ConstTensor; cellToOutputWeights:ConstTensor; + + inputLayerNormWeights:ConstTensor; + forgetLayerNormWeights:ConstTensor; + cellLayerNormWeights:ConstTensor; + outputLayerNormWeights:ConstTensor; } table LstmDescriptor { @@ -546,6 +551,7 @@ table LstmDescriptor { cifgEnabled:bool = true; peepholeEnabled:bool = false; projectionEnabled:bool = false; + layerNormEnabled:bool = false; } table LstmLayer { |