1 files changed, 22 insertions, 0 deletions
diff --git a/tests/dataset/ShapeDatasets.h b/tests/dataset/ShapeDatasets.h
index 73bdb8ea0e..d2b82cae40 100644
--- a/tests/dataset/ShapeDatasets.h
+++ b/tests/dataset/ShapeDatasets.h
@@ -102,6 +102,17 @@ public:
     }
 };
 
+/** Data set containing two small 2D tensor shapes. */
+class Small2DShapes final : public ShapeDataset<2>
+{
+public:
+    Small2DShapes()
+        : ShapeDataset(TensorShape(5U, 5U),
+                       TensorShape(640U, 480U))
+    {
+    }
+};
+
 /** Data set containing small tensor shapes. */
 class SmallShapes final : public ShapeDataset<3>
 {
@@ -125,6 +136,17 @@ public:
     {
     }
 };
+
+/** Data set containing two 2D large tensor shapes. */
+class Large2DShapes final : public ShapeDataset<2>
+{
+public:
+    Large2DShapes()
+        : ShapeDataset(TensorShape(1920U, 1080U),
+                       TensorShape(4160U, 3120U))
+    {
+    }
+};
 } // namespace test
 } // namespace arm_compute
 #endif