diff options
Diffstat (limited to 'tests/validation/NEON/LSTMLayer.cpp')
| -rw-r--r-- | tests/validation/NEON/LSTMLayer.cpp | 15 |
1 files changed, 9 insertions, 6 deletions
diff --git a/tests/validation/NEON/LSTMLayer.cpp b/tests/validation/NEON/LSTMLayer.cpp index b27dfae8fa..c503972ba9 100644 --- a/tests/validation/NEON/LSTMLayer.cpp +++ b/tests/validation/NEON/LSTMLayer.cpp @@ -153,10 +153,11 @@ template <typename T> using NELSTMLayerFixture = LSTMLayerValidationFixture<Tensor, Accessor, NELSTMLayer, LSTMParams<ITensor>, T>; TEST_SUITE(FP32) -FIXTURE_DATA_TEST_CASE(RunSmall, NELSTMLayerFixture<float>, framework::DatasetMode::ALL, combine(combine(combine(datasets::SmallLSTMLayerDataset(), framework::dataset::make("DataType", +FIXTURE_DATA_TEST_CASE(RunSmall, NELSTMLayerFixture<float>, framework::DatasetMode::ALL, combine(combine(combine(combine(datasets::SmallLSTMLayerDataset(), framework::dataset::make("DataType", DataType::F32)), - framework::dataset::make("ProjectionOpt", { true, false })), - framework::dataset::make("PeepholeOpt", { true, false }))) + framework::dataset::make("ProjectionOpt", { true, false })), + framework::dataset::make("PeepholeOpt", { true, false })), + framework::dataset::make("UseLayerNorm", { false }))) { // Validate output validate(Accessor(_target), _reference, tolerance_f32); @@ -166,9 +167,11 @@ TEST_SUITE_END() // FP32 #ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC TEST_SUITE(FP16) -FIXTURE_DATA_TEST_CASE(RunSmall, NELSTMLayerFixture<half>, framework::DatasetMode::ALL, combine(combine(combine(datasets::SmallLSTMLayerDataset(), framework::dataset::make("DataType", DataType::F16)), - framework::dataset::make("ProjectionOpt", { true, false })), - framework::dataset::make("PeepholeOpt", { true, false }))) +FIXTURE_DATA_TEST_CASE(RunSmall, NELSTMLayerFixture<half>, framework::DatasetMode::ALL, combine(combine(combine(combine(datasets::SmallLSTMLayerDataset(), framework::dataset::make("DataType", + DataType::F16)), + framework::dataset::make("ProjectionOpt", { true, false })), + framework::dataset::make("PeepholeOpt", { true, false })), + framework::dataset::make("UseLayerNorm", { false }))) { // Validate output validate(Accessor(_target), _reference, tolerance_f16); |