aboutsummaryrefslogtreecommitdiff
path: root/tests/datasets/ShapeDatasets.h
diff options
context:
space:
mode:
authorGian Marco Iodice <gianmarco.iodice@arm.com>2018-09-13 15:56:47 +0100
committerAnthony Barbier <anthony.barbier@arm.com>2018-11-02 16:54:54 +0000
commit8e1f3c0685a4a0413c6d57a3254b5d59949ec255 (patch)
treeea2124c96b4c49099eaa05a1b3751aad0a49cf83 /tests/datasets/ShapeDatasets.h
parent8bab0ee5f44a0e2cfe60d4d9e040a2f5ae4ef9b1 (diff)
downloadComputeLibrary-8e1f3c0685a4a0413c6d57a3254b5d59949ec255.tar.gz
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>
Diffstat (limited to 'tests/datasets/ShapeDatasets.h')
-rw-r--r--tests/datasets/ShapeDatasets.h3
1 files changed, 1 insertions, 2 deletions
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 },
})
{
}