diff options
| author | Michele Di Giorgio <michele.digiorgio@arm.com> | 2019-06-13 17:01:29 +0100 |
|---|---|---|
| committer | Michele Di Giorgio <michele.digiorgio@arm.com> | 2019-06-19 11:10:57 +0000 |
| commit | 0cbfda629dd8f684e625173341bab972f004222c (patch) | |
| tree | d3a05f625b9564e8f8977a8151ba5ebc335960a0 /tests | |
| parent | 75bde5e21cfbf5e699a3a89655d97fec7c0892e7 (diff) | |
| download | ComputeLibrary-0cbfda629dd8f684e625173341bab972f004222c.tar.gz | |
COMPMID-2343: Add layer normalization support in NELSTMLayer
Change-Id: I1f620d70c6eaadfb9e3a1b345de350ac0253b65c
Signed-off-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1366
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Manuel Bottini <manuel.bottini@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/validation/NEON/LSTMLayer.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
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<float>, 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<half>, 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); |