COMPMID-415 - Fixed bad_alloc in DepthConcatenate reducing the size of input tensor for the NIGHTLY tests

Change-Id: I06a4b3f56c5157a67548570371d78f8a73029c2d
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/87601
Reviewed-by: Pablo Tello <pablo.tello@arm.com>
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>

1 files changed, 16 insertions, 0 deletions

diff --git a/tests/datasets/ShapeDatasets.h b/tests/datasets/ShapeDatasets.h
index 806fc04c0d..ddcf6a7a95 100644
--- a/tests/datasets/ShapeDatasets.h
+++ b/tests/datasets/ShapeDatasets.h
@@ -202,6 +202,22 @@ public:
     {
     }
 };
+
+/** Data set containing 2D tensor shapes for DepthConcatenate. */
+class DepthConcatenateShapes final : public ShapeDataset
+{
+public:
+    DepthConcatenateShapes()
+        : ShapeDataset("Shape",
+    {
+        TensorShape{ 322U, 243U },
+                     TensorShape{ 463U, 879U },
+                     TensorShape{ 416U, 651U }
+    })
+    {
+    }
+};
+
 } // namespace datasets
 } // namespace test
 } // namespace arm_compute