diff options
Diffstat (limited to 'tests/dataset')
-rw-r--r-- | tests/dataset/ShapeDatasets.h | 22 |
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 |