From e96e4f0e49a3571f823927913dde173513baee72 Mon Sep 17 00:00:00 2001 From: Pablo Tello Date: Fri, 21 Dec 2018 16:47:23 +0000 Subject: COMPMID-1766: Implemented CPP Non Max Suppression Change-Id: I2d2b684d464f7b3bb1f91cfd29952f612d65f11f Signed-off-by: Pablo Tello Reviewed-on: https://review.mlplatform.org/708 Reviewed-by: VidhyaSudhan Loganathan Tested-by: Arm Jenkins --- tests/datasets/ShapeDatasets.h | 31 +++++++++++++++++++++++++++++++ 1 file changed, 31 insertions(+) (limited to 'tests/datasets') diff --git a/tests/datasets/ShapeDatasets.h b/tests/datasets/ShapeDatasets.h index 480df3c7eb..f461d7f9d2 100644 --- a/tests/datasets/ShapeDatasets.h +++ b/tests/datasets/ShapeDatasets.h @@ -946,6 +946,37 @@ public: { } }; + +/** Data set containing small 2D tensor shapes. */ +class Small2DNonMaxSuppressionShapes final : public ShapeDataset +{ +public: + Small2DNonMaxSuppressionShapes() + : ShapeDataset("Shape", + { + TensorShape{ 4U, 7U }, + TensorShape{ 4U, 13U }, + TensorShape{ 4U, 64U } + }) + { + } +}; + +/** Data set containing large 2D tensor shapes. */ +class Large2DNonMaxSuppressionShapes final : public ShapeDataset +{ +public: + Large2DNonMaxSuppressionShapes() + : ShapeDataset("Shape", + { + TensorShape{ 4U, 207U }, + TensorShape{ 4U, 113U }, + TensorShape{ 4U, 264U } + }) + { + } +}; + } // namespace datasets } // namespace test } // namespace arm_compute -- cgit v1.2.1