From 8e1f3c0685a4a0413c6d57a3254b5d59949ec255 Mon Sep 17 00:00:00 2001 From: Gian Marco Iodice Date: Thu, 13 Sep 2018 15:56:47 +0100 Subject: COMPMID-1451 - Reduced dataset for YOLOLayer Change-Id: I97c5ccdc33c16eaf0d90ac0a1ec6a066fc550842 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/148320 Reviewed-by: Georgios Pinitas Tested-by: bsgcomp --- tests/datasets/ShapeDatasets.h | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) (limited to 'tests/datasets/ShapeDatasets.h') diff --git a/tests/datasets/ShapeDatasets.h b/tests/datasets/ShapeDatasets.h index 551e7ffa8c..5ec3d4c5cb 100644 --- a/tests/datasets/ShapeDatasets.h +++ b/tests/datasets/ShapeDatasets.h @@ -857,8 +857,7 @@ public: // Batch size 1 TensorShape{ 11U, 11U, 270U }, TensorShape{ 27U, 13U, 90U }, - TensorShape{ 128U, 64U, 45U, 2U }, - TensorShape{ 11U, 11U, 45U, 3U } + TensorShape{ 13U, 12U, 45U, 2U }, }) { } -- cgit v1.2.1