diff options
author | Moritz Pflanzer <moritz.pflanzer@arm.com> | 2017-07-26 11:49:37 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-09-17 14:16:42 +0100 |
commit | b3d2579b567eabd98fdb4861bf1380fefa18c9aa (patch) | |
tree | ec684767bf8e445162abb2b372cce46dc6d16443 /tests/datasets_new/ShapeDatasets.h | |
parent | 8594b1139fd72e541e725296bd8bb625496f3381 (diff) | |
download | ComputeLibrary-b3d2579b567eabd98fdb4861bf1380fefa18c9aa.tar.gz |
COMPMID-415: Move ConvolutionLayer to new validation
Change-Id: I1f40dff43142c4e2c096122bfa1ca08241ff80ff
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/81952
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'tests/datasets_new/ShapeDatasets.h')
-rw-r--r-- | tests/datasets_new/ShapeDatasets.h | 21 |
1 files changed, 18 insertions, 3 deletions
diff --git a/tests/datasets_new/ShapeDatasets.h b/tests/datasets_new/ShapeDatasets.h index ba142cae0c..14f7851621 100644 --- a/tests/datasets_new/ShapeDatasets.h +++ b/tests/datasets_new/ShapeDatasets.h @@ -35,7 +35,7 @@ namespace test { namespace datasets { -/** Data set containing one 1D tensor shape. */ +/** Data set containing 1D tensor shapes. */ class Small1DShape final : public framework::dataset::SingletonDataset<TensorShape> { public: @@ -48,7 +48,7 @@ public: /** Parent type for all for shape datasets. */ using ShapeDataset = framework::dataset::ContainerDataset<std::vector<TensorShape>>; -/** Data set containing two small 2D tensor shapes. */ +/** Data set containing small 2D tensor shapes. */ class Small2DShapes final : public ShapeDataset { public: @@ -93,7 +93,7 @@ public: } }; -/** Data set containing two 2D large tensor shapes. */ +/** Data set containing large 2D tensor shapes. */ class Large2DShapes final : public ShapeDataset { public: @@ -107,6 +107,21 @@ public: { } }; + +/** Data set containing small tensor shapes for direct convolution. */ +class SmallDirectConvolutionShapes final : public ShapeDataset +{ +public: + SmallDirectConvolutionShapes() + : ShapeDataset("InputShape", + { + TensorShape{ 3U, 3U, 3U, 2U, 4U, 5U }, + TensorShape{ 32U, 37U, 3U }, + TensorShape{ 13U, 15U, 8U, 3U } + }) + { + } +}; } // namespace datasets } // namespace test } // namespace arm_compute |