From 37d65e4fd036529a2c7567f1987235d0b0258ab0 Mon Sep 17 00:00:00 2001 From: SiCong Li Date: Fri, 6 Nov 2020 09:55:04 +0000 Subject: 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 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/4393 Comments-Addressed: Arm Jenkins Reviewed-by: Michele Di Giorgio Reviewed-by: Georgios Pinitas Tested-by: Arm Jenkins --- tests/datasets/ShapeDatasets.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'tests/datasets/ShapeDatasets.h') diff --git a/tests/datasets/ShapeDatasets.h b/tests/datasets/ShapeDatasets.h index a7f1a44286..558b37af33 100644 --- a/tests/datasets/ShapeDatasets.h +++ b/tests/datasets/ShapeDatasets.h @@ -839,7 +839,7 @@ public: SoftmaxLayerSmallShapes() : ShapeDataset("Shape", { - TensorShape{ 9U, 9U }, + TensorShape{ 1U, 9U }, TensorShape{ 256U, 10U }, TensorShape{ 353U, 8U }, TensorShape{ 781U, 5U }, -- cgit v1.2.1