COMPMID-761: Add CL/NEON Magnitude benchmark tests

Change-Id: I9ed3718679d4bc96300a23ce8063d5e12c201bf9
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/114166
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>

1 files changed, 29 insertions, 0 deletions

diff --git a/tests/datasets/ShapeDatasets.h b/tests/datasets/ShapeDatasets.h
index c9e5510760..58fba07bf8 100644
--- a/tests/datasets/ShapeDatasets.h
+++ b/tests/datasets/ShapeDatasets.h
@@ -337,6 +337,35 @@ public:
     }
 };
 
+/** Data set containing 2D tensor shapes relative to an image size. */
+class SmallImageShapes final : public ShapeDataset
+{
+public:
+    SmallImageShapes()
+        : ShapeDataset("Shape",
+    {
+        TensorShape{ 640U, 480U },
+                     TensorShape{ 800U, 600U },
+                     TensorShape{ 1200U, 800U }
+    })
+    {
+    }
+};
+
+/** Data set containing 2D tensor shapes relative to an image size. */
+class LargeImageShapes final : public ShapeDataset
+{
+public:
+    LargeImageShapes()
+        : ShapeDataset("Shape",
+    {
+        TensorShape{ 1920U, 1080U },
+                     TensorShape{ 2560U, 1536U },
+                     TensorShape{ 3584U, 2048U }
+    })
+    {
+    }
+};
 } // namespace datasets
 } // namespace test
 } // namespace arm_compute