diff options
author | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2018-12-06 17:13:09 +0000 |
---|---|---|
committer | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2018-12-10 15:58:54 +0000 |
commit | 5ba5e0938e68d4f90f5545a81066d56f022b376a (patch) | |
tree | d828f8b3fd52e6d5b8f732a7ec41f832f0b921d8 /tests/datasets | |
parent | 1d7cbb99d2a34abd15f3b6c2e017115736cd90cc (diff) | |
download | ComputeLibrary-5ba5e0938e68d4f90f5545a81066d56f022b376a.tar.gz |
COMPMID-1774: Implement CLGEMMReshapeLHSMatrixKernel to reshape the LHS matrix of GEMM/GEMMLowp
Change-Id: I8c5fd4c8bcdffda1522c83158981ed92baa045f4
Reviewed-on: https://review.mlplatform.org/364
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'tests/datasets')
-rw-r--r-- | tests/datasets/ShapeDatasets.h | 56 |
1 files changed, 56 insertions, 0 deletions
diff --git a/tests/datasets/ShapeDatasets.h b/tests/datasets/ShapeDatasets.h index f7e7ae26b9..875f7e8ea1 100644 --- a/tests/datasets/ShapeDatasets.h +++ b/tests/datasets/ShapeDatasets.h @@ -892,6 +892,62 @@ public: { } }; + +/** Data set containing small tensor shapes to be used with the GEMM reshaping kernel */ +class SmallGEMMReshape2DShapes final : public ShapeDataset +{ +public: + SmallGEMMReshape2DShapes() + : ShapeDataset("Shape", + { + TensorShape{ 63U, 72U }, + }) + { + } +}; + +/** Data set containing small tensor shapes to be used with the GEMM reshaping kernel when the input has to be reinterpreted as 3D */ +class SmallGEMMReshape3DShapes final : public ShapeDataset +{ +public: + SmallGEMMReshape3DShapes() + : ShapeDataset("Shape", + { + TensorShape{ 63U, 9U, 8U }, + }) + { + } +}; + +/** Data set containing large tensor shapes to be used with the GEMM reshaping kernel */ +class LargeGEMMReshape2DShapes final : public ShapeDataset +{ +public: + LargeGEMMReshape2DShapes() + : ShapeDataset("Shape", + { + TensorShape{ 16U, 27U }, + TensorShape{ 533U, 171U }, + TensorShape{ 345U, 612U } + }) + { + } +}; + +/** Data set containing large tensor shapes to be used with the GEMM reshaping kernel when the input has to be reinterpreted as 3D */ +class LargeGEMMReshape3DShapes final : public ShapeDataset +{ +public: + LargeGEMMReshape3DShapes() + : ShapeDataset("Shape", + { + TensorShape{ 16U, 3U, 9U }, + TensorShape{ 533U, 19U, 9U }, + TensorShape{ 345U, 34U, 18U } + }) + { + } +}; } // namespace datasets } // namespace test } // namespace arm_compute |