aboutsummaryrefslogtreecommitdiff
path: root/tests/datasets/RandomBatchNormalizationLayerDataset.h
diff options
context:
space:
mode:
authorSiCong Li <sicong.li@arm.com>2020-11-06 09:55:04 +0000
committerGeorgios Pinitas <georgios.pinitas@arm.com>2021-04-07 14:14:23 +0000
commit37d65e4fd036529a2c7567f1987235d0b0258ab0 (patch)
treecf1166b9af7366c4b06d13d5556196032f585291 /tests/datasets/RandomBatchNormalizationLayerDataset.h
parentc5b1beec99d2be7ff3babbc6345b1baa81f29eb7 (diff)
downloadComputeLibrary-37d65e4fd036529a2c7567f1987235d0b0258ab0.tar.gz
Review all shapes in datasets to account for padding removal Part 2
* Fix validate compare_dimensions to account for cases where 1D tensor permuted into a 3D tensor * Add the following configurations to stress padding removal: * size = 1 * size = multiple of processing size * size = non-multiple of processing size Partially resolves COMPMID-3865 Change-Id: Iee0de4c9e72b3413c0807e2c86bc2331911a4c6f Signed-off-by: SiCong Li <sicong.li@arm.com> Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/4393 Comments-Addressed: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com> Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com> Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'tests/datasets/RandomBatchNormalizationLayerDataset.h')
-rw-r--r--tests/datasets/RandomBatchNormalizationLayerDataset.h6
1 files changed, 3 insertions, 3 deletions
diff --git a/tests/datasets/RandomBatchNormalizationLayerDataset.h b/tests/datasets/RandomBatchNormalizationLayerDataset.h
index 5a49dd702b..4ccb2eaaba 100644
--- a/tests/datasets/RandomBatchNormalizationLayerDataset.h
+++ b/tests/datasets/RandomBatchNormalizationLayerDataset.h
@@ -1,5 +1,5 @@
/*
- * Copyright (c) 2017-2019 Arm Limited.
+ * Copyright (c) 2017-2021 Arm Limited.
*
* SPDX-License-Identifier: MIT
*
@@ -42,9 +42,9 @@ class SmallRandomBatchNormalizationLayerDataset final : public BatchNormalizatio
public:
SmallRandomBatchNormalizationLayerDataset()
{
- add_config(TensorShape(15U, 16U, 2U, 12U), TensorShape(2U), 0.1f);
+ add_config(TensorShape(1U, 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);
+ add_config(TensorShape(32U, 3U, 6U, 11U), TensorShape(6U), 0.1f);
}
};
class LargeRandomBatchNormalizationLayerDataset final : public BatchNormalizationLayerDataset