aboutsummaryrefslogtreecommitdiff
path: root/tests/datasets/ShapeDatasets.h
diff options
context:
space:
mode:
authorMichalis Spyrou <michalis.spyrou@arm.com>2018-05-09 09:59:23 +0100
committerAnthony Barbier <anthony.barbier@arm.com>2018-11-02 16:52:35 +0000
commit55b3d1216b4011d86d5f06335e518dc924987ae5 (patch)
treea94ca420f4385ff301a770fbeb5b4f930ac60105 /tests/datasets/ShapeDatasets.h
parenta8aef2916379402e241d9f2c5e0faf3f99c860f7 (diff)
downloadComputeLibrary-55b3d1216b4011d86d5f06335e518dc924987ae5.tar.gz
COMPMID-1137 OpenCL concatenate width
Change-Id: I40faba421281b1cf080fa6a825d04a4366cdaeb0 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/130700 Reviewed-by: Anthony Barbier <anthony.barbier@arm.com> Tested-by: Jenkins <bsgcomp@arm.com> Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Diffstat (limited to 'tests/datasets/ShapeDatasets.h')
-rw-r--r--tests/datasets/ShapeDatasets.h15
1 files changed, 15 insertions, 0 deletions
diff --git a/tests/datasets/ShapeDatasets.h b/tests/datasets/ShapeDatasets.h
index 76cf9a1835..b138a425b9 100644
--- a/tests/datasets/ShapeDatasets.h
+++ b/tests/datasets/ShapeDatasets.h
@@ -523,6 +523,21 @@ public:
}
};
+/** Data set containing tensor shapes for WidthConcatenateLayer. */
+class WidthConcatenateLayerShapes final : public ShapeDataset
+{
+public:
+ WidthConcatenateLayerShapes()
+ : ShapeDataset("Shape",
+ {
+ TensorShape{ 232U, 65U, 3U },
+ TensorShape{ 432U, 65U, 3U },
+ TensorShape{ 124U, 65U, 3U }
+ })
+ {
+ }
+};
+
/** Data set containing global pooling tensor shapes. */
class GlobalPoolingShapes final : public ShapeDataset
{