From 5daeffdd96c5e46ac2482431d578dfaf9b300cde Mon Sep 17 00:00:00 2001 From: Michele Di Giorgio Date: Mon, 26 Nov 2018 10:01:15 +0000 Subject: COMPMID-1723: CL: Implement Reverse Change-Id: Id0d4a07af24e2331161996083b0c1bab072bd405 Reviewed-on: https://review.mlplatform.org/322 Reviewed-by: Georgios Pinitas Tested-by: Arm Jenkins --- tests/datasets/ShapeDatasets.h | 14 ++++++++++++++ 1 file changed, 14 insertions(+) (limited to 'tests/datasets') diff --git a/tests/datasets/ShapeDatasets.h b/tests/datasets/ShapeDatasets.h index ffacf34620..f7e7ae26b9 100644 --- a/tests/datasets/ShapeDatasets.h +++ b/tests/datasets/ShapeDatasets.h @@ -38,6 +38,20 @@ namespace datasets /** Parent type for all for shape datasets. */ using ShapeDataset = framework::dataset::ContainerDataset>; +/** Data set containing tiny 1D tensor shapes. */ +class Tiny1DShapes final : public ShapeDataset +{ +public: + Tiny1DShapes() + : ShapeDataset("Shape", + { + TensorShape{ 2U }, + TensorShape{ 3U }, + }) + { + } +}; + /** Data set containing small 1D tensor shapes. */ class Small1DShapes final : public ShapeDataset { -- cgit v1.2.1