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>
diff --git a/tests/datasets/ShapeDatasets.h b/tests/datasets/ShapeDatasets.h
index c9e5510..58fba07 100644
--- a/tests/datasets/ShapeDatasets.h
+++ b/tests/datasets/ShapeDatasets.h
@@ -337,6 +337,35 @@
}
};
+/** 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