From 0cbfda629dd8f684e625173341bab972f004222c Mon Sep 17 00:00:00 2001 From: Michele Di Giorgio Date: Thu, 13 Jun 2019 17:01:29 +0100 Subject: COMPMID-2343: Add layer normalization support in NELSTMLayer Change-Id: I1f620d70c6eaadfb9e3a1b345de350ac0253b65c Signed-off-by: Michele Di Giorgio Reviewed-on: https://review.mlplatform.org/c/1366 Tested-by: Arm Jenkins Reviewed-by: Manuel Bottini Comments-Addressed: Arm Jenkins Reviewed-by: Georgios Pinitas --- tests/validation/NEON/LSTMLayer.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'tests/validation/NEON/LSTMLayer.cpp') diff --git a/tests/validation/NEON/LSTMLayer.cpp b/tests/validation/NEON/LSTMLayer.cpp index c503972ba9..45beb36e60 100644 --- a/tests/validation/NEON/LSTMLayer.cpp +++ b/tests/validation/NEON/LSTMLayer.cpp @@ -157,7 +157,7 @@ FIXTURE_DATA_TEST_CASE(RunSmall, NELSTMLayerFixture, framework::DatasetMo DataType::F32)), framework::dataset::make("ProjectionOpt", { true, false })), framework::dataset::make("PeepholeOpt", { true, false })), - framework::dataset::make("UseLayerNorm", { false }))) + framework::dataset::make("UseLayerNorm", { true, false }))) { // Validate output validate(Accessor(_target), _reference, tolerance_f32); @@ -171,7 +171,7 @@ FIXTURE_DATA_TEST_CASE(RunSmall, NELSTMLayerFixture, framework::DatasetMod DataType::F16)), framework::dataset::make("ProjectionOpt", { true, false })), framework::dataset::make("PeepholeOpt", { true, false })), - framework::dataset::make("UseLayerNorm", { false }))) + framework::dataset::make("UseLayerNorm", { true, false }))) { // Validate output validate(Accessor(_target), _reference, tolerance_f16); -- cgit v1.2.1