From f8c629701e760e2476582c91e6b7f5e1313dc02a Mon Sep 17 00:00:00 2001
From: Jan Eilers <jan.eilers@arm.com>
Date: Wed, 17 Jul 2019 11:07:49 +0100
Subject: 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 | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'src/armnnSerializer/ArmnnSchema.fbs')

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 {
-- 
cgit v1.2.1