diff options
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 { |