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>
diff --git a/tests/datasets/ShapeDatasets.h b/tests/datasets/ShapeDatasets.h
index 806fc04..ddcf6a7 100644
--- a/tests/datasets/ShapeDatasets.h
+++ b/tests/datasets/ShapeDatasets.h
@@ -202,6 +202,22 @@
{
}
};
+
+/** 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