commit 8e1f3c0685a4a0413c6d57a3254b5d59949ec255
author Gian Marco Iodice <gianmarco.iodice@arm.com> Thu Sep 13 15:56:47 2018 +0100
committer Anthony Barbier <anthony.barbier@arm.com> Fri Nov 02 16:54:54 2018 +0000
tree ea2124c96b4c49099eaa05a1b3751aad0a49cf83
parent 8bab0ee5f44a0e2cfe60d4d9e040a2f5ae4ef9b1

COMPMID-1451 - Reduced dataset for YOLOLayer

Change-Id: I97c5ccdc33c16eaf0d90ac0a1ec6a066fc550842
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/148320
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Tested-by: bsgcomp <bsgcomp@arm.com>

tests/datasets/ShapeDatasets.h

diff --git a/tests/datasets/ShapeDatasets.h b/tests/datasets/ShapeDatasets.h
index 551e7ff..5ec3d4c 100644
--- a/tests/datasets/ShapeDatasets.h
+++ b/tests/datasets/ShapeDatasets.h
@@ -857,8 +857,7 @@
         // 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 },
     })
     {
     }