From 7e4b23953e885e58d655a7d9f35a1afcc38365e4 Mon Sep 17 00:00:00 2001 From: Gian Marco Iodice Date: Thu, 22 Feb 2018 16:17:20 +0000 Subject: COMPMID-935 - Implementing Convolution with Winograd on OpenCL (part 2) Implemented Winograd Filter Transform 3x3 on OpenCL Change-Id: I8f2b2dd938c5c000ef7ce392a37fb7b8b4202a4e Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/122708 Reviewed-by: Georgios Pinitas Tested-by: Jenkins --- tests/datasets/ShapeDatasets.h | 32 ++++++++++++++++++++++++++++++++ 1 file changed, 32 insertions(+) (limited to 'tests/datasets/ShapeDatasets.h') diff --git a/tests/datasets/ShapeDatasets.h b/tests/datasets/ShapeDatasets.h index 4b563708e1..e939a6f5a7 100644 --- a/tests/datasets/ShapeDatasets.h +++ b/tests/datasets/ShapeDatasets.h @@ -238,6 +238,38 @@ public: } }; +/** Data set containing medium 3D tensor shapes. */ +class Medium3DShapes final : public ShapeDataset +{ +public: + Medium3DShapes() + : ShapeDataset("Shape", + { + TensorShape{ 42U, 37U, 8U }, + TensorShape{ 57U, 60U, 13U }, + TensorShape{ 128U, 64U, 21U }, + TensorShape{ 83U, 72U, 14U } + }) + { + } +}; + +/** Data set containing medium 4D tensor shapes. */ +class Medium4DShapes final : public ShapeDataset +{ +public: + Medium4DShapes() + : ShapeDataset("Shape", + { + TensorShape{ 42U, 37U, 8U, 15U }, + TensorShape{ 57U, 60U, 13U, 8U }, + TensorShape{ 128U, 64U, 21U, 13U }, + TensorShape{ 83U, 72U, 14U, 5U } + }) + { + } +}; + /** Data set containing large tensor shapes. */ class LargeShapes final : public ShapeDataset { -- cgit v1.2.1