aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer/ArmnnSchema.fbs
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnnSerializer/ArmnnSchema.fbs')
-rw-r--r--src/armnnSerializer/ArmnnSchema.fbs6
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 {