diff options
author | Michalis Spyrou <michalis.spyrou@arm.com> | 2019-01-09 14:21:03 +0000 |
---|---|---|
committer | Michalis Spyrou <michalis.spyrou@arm.com> | 2019-01-11 12:32:03 +0000 |
commit | aeebe4ad332bc6dd4797243b39ec6000975429c5 (patch) | |
tree | 68866e55aa29ad860ee9d222869b4add2551619b /tests/datasets/RandomBatchNormalizationLayerDataset.h | |
parent | ee7c15d8a57b6e1a0a98edf2bb4693024d9c15dd (diff) | |
download | ComputeLibrary-aeebe4ad332bc6dd4797243b39ec6000975429c5.tar.gz |
COMPMID-1652 NEON Cleanup and add missing tests
Change-Id: I7b351f18a78ed8a250bf3a91ef320db61984146a
Reviewed-on: https://review.mlplatform.org/485
Reviewed-by: Isabella Gottardi <isabella.gottardi@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Diffstat (limited to 'tests/datasets/RandomBatchNormalizationLayerDataset.h')
-rw-r--r-- | tests/datasets/RandomBatchNormalizationLayerDataset.h | 15 |
1 files changed, 12 insertions, 3 deletions
diff --git a/tests/datasets/RandomBatchNormalizationLayerDataset.h b/tests/datasets/RandomBatchNormalizationLayerDataset.h index 10908871b6..2acce6ee04 100644 --- a/tests/datasets/RandomBatchNormalizationLayerDataset.h +++ b/tests/datasets/RandomBatchNormalizationLayerDataset.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017 ARM Limited. + * Copyright (c) 2017-2019 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -37,16 +37,25 @@ namespace test { namespace datasets { -class RandomBatchNormalizationLayerDataset final : public BatchNormalizationLayerDataset +class SmallRandomBatchNormalizationLayerDataset final : public BatchNormalizationLayerDataset { public: - RandomBatchNormalizationLayerDataset() + SmallRandomBatchNormalizationLayerDataset() { add_config(TensorShape(15U, 16U, 2U, 12U), TensorShape(2U), 0.1f); add_config(TensorShape(21U, 11U, 12U, 7U), TensorShape(12U), 0.1f); add_config(TensorShape(7U, 3U, 6U, 11U), TensorShape(6U), 0.1f); } }; +class LargeRandomBatchNormalizationLayerDataset final : public BatchNormalizationLayerDataset +{ +public: + LargeRandomBatchNormalizationLayerDataset() + { + add_config(TensorShape(111U, 47U, 21U, 11U), TensorShape(21U), 0.1f); + add_config(TensorShape(236U, 3U, 169U, 7U), TensorShape(169U), 0.1f); + } +}; } // namespace datasets } // namespace test } // namespace arm_compute |