From 17d6d3cb0f92517b3764870b85cb7ffd22332556 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Mon, 2 Jul 2018 17:52:40 +0100 Subject: COMPMID-1352: Disable support for 4D softmax layer. Change-Id: Ia8afabb36e644895d321ded51a6a0676347443e1 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/138387 Reviewed-by: Gian Marco Iodice Tested-by: Jenkins --- tests/datasets/ShapeDatasets.h | 13 +++++-------- 1 file changed, 5 insertions(+), 8 deletions(-) (limited to 'tests/datasets') diff --git a/tests/datasets/ShapeDatasets.h b/tests/datasets/ShapeDatasets.h index 68263c7793..766530b0d7 100644 --- a/tests/datasets/ShapeDatasets.h +++ b/tests/datasets/ShapeDatasets.h @@ -627,7 +627,7 @@ public: : ShapeDataset("Shape", { TensorShape{ 9U, 9U }, - TensorShape{ 128U, 10U, 2U }, + TensorShape{ 128U, 10U }, }) { } @@ -641,11 +641,9 @@ public: : ShapeDataset("Shape", { TensorShape{ 9U, 9U }, - TensorShape{ 256U, 10U, 2U }, - TensorShape{ 353U, 8U, 2U, 2U }, - TensorShape{ 512U, 7U, 2U, 2U }, - TensorShape{ 633U, 10U, 1U, 2U }, - TensorShape{ 781U, 5U, 2U }, + TensorShape{ 256U, 10U }, + TensorShape{ 353U, 8U }, + TensorShape{ 781U, 5U }, }) { } @@ -659,8 +657,7 @@ public: : ShapeDataset("Shape", { TensorShape{ 1000U, 10U }, - TensorShape{ 3989U, 10U, 2U }, - TensorShape{ 4098U, 8U, 1U, 2U }, + TensorShape{ 3989U, 10U }, TensorShape{ 7339U, 11U }, }) { -- cgit v1.2.1