commit f8c629701e760e2476582c91e6b7f5e1313dc02a
author Jan Eilers <jan.eilers@arm.com> Wed Jul 17 11:07:49 2019 +0100
committer Jan Eilers <jan.eilers@arm.com> Tue Jul 23 11:01:29 2019 +0100
tree 3c9dc1a673fe99367a7f25d98199d8ab386e8bc2
parent 07f2121feeeeae36a7e67eeb8a6965df63b848f3

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>

src/armnnSerializer/ArmnnSchema.fbs

diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs
index e3953df..0fd8da7 100644
--- a/src/armnnSerializer/ArmnnSchema.fbs
+++ b/src/armnnSerializer/ArmnnSchema.fbs
@@ -537,6 +537,11 @@
 
     cellToForgetWeights:ConstTensor;
     cellToOutputWeights:ConstTensor;
+
+    inputLayerNormWeights:ConstTensor;
+    forgetLayerNormWeights:ConstTensor;
+    cellLayerNormWeights:ConstTensor;
+    outputLayerNormWeights:ConstTensor;
 }
 
 table LstmDescriptor {
@@ -546,6 +551,7 @@
     cifgEnabled:bool = true;
     peepholeEnabled:bool = false;
     projectionEnabled:bool = false;
+    layerNormEnabled:bool = false;
 }
 
 table LstmLayer {