From 02ee4291795f64fb510a71c6c754671438635186 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Thu, 15 Feb 2018 17:22:36 +0000 Subject: COMPMID-765: Fix CPPPermute error when permuting the strides. Change-Id: I4ea57579d997dd6a2e248634e3b7cb58bb3e2838 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/120693 Reviewed-by: Pablo Tello Tested-by: Jenkins --- tests/datasets/ShapeDatasets.h | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) (limited to 'tests/datasets/ShapeDatasets.h') diff --git a/tests/datasets/ShapeDatasets.h b/tests/datasets/ShapeDatasets.h index 733394224c..4b563708e1 100644 --- a/tests/datasets/ShapeDatasets.h +++ b/tests/datasets/ShapeDatasets.h @@ -100,7 +100,8 @@ public: Small3DShapes() : ShapeDataset("Shape", { - TensorShape{ 7U, 7U, 5U }, + TensorShape{ 1U, 7U, 7U }, + TensorShape{ 7U, 7U, 5U }, TensorShape{ 27U, 13U, 37U }, TensorShape{ 128U, 64U, 21U } }) @@ -128,7 +129,8 @@ public: Small4DShapes() : ShapeDataset("Shape", { - TensorShape{ 7U, 7U, 5U, 3U }, + TensorShape{ 1U, 7U, 1U, 3U }, + TensorShape{ 7U, 7U, 5U, 3U }, TensorShape{ 27U, 13U, 37U, 2U }, TensorShape{ 128U, 64U, 21U, 3U } }) -- cgit v1.2.1