diff options
author | Chunosov <N.Chunosov@yandex.ru> | 2017-11-06 22:09:45 +0700 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:35:24 +0000 |
commit | d6afedc775220f17317f1835a4d18b72a54525de (patch) | |
tree | 54aed8322a4a286ba376d74bbee61c85a588cc9b /tests/datasets/ShapeDatasets.h | |
parent | 6ff12a0f7765f62b8d0fa8554021e1cac2789f19 (diff) | |
download | ComputeLibrary-d6afedc775220f17317f1835a4d18b72a54525de.tar.gz |
COMPMID-661: softmax-fp32 optimisation (#14)
Change-Id: I2007af1ed9dcf68065cf412aa50f73a2025b31a6
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/94605
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Diffstat (limited to 'tests/datasets/ShapeDatasets.h')
-rw-r--r-- | tests/datasets/ShapeDatasets.h | 34 |
1 files changed, 34 insertions, 0 deletions
diff --git a/tests/datasets/ShapeDatasets.h b/tests/datasets/ShapeDatasets.h index 86ed2b2ad7..45f5d1c9ff 100644 --- a/tests/datasets/ShapeDatasets.h +++ b/tests/datasets/ShapeDatasets.h @@ -269,6 +269,40 @@ public: } }; +/** Data set containing small softmax layer shapes. */ +class SoftmaxLayerSmallShapes final : public ShapeDataset +{ +public: + SoftmaxLayerSmallShapes() + : ShapeDataset("Shape", + { + TensorShape{ 9U, 9U }, + TensorShape{ 256U, 10U, 2U }, + TensorShape{ 353U, 8U, 2U, 2U }, + TensorShape{ 512U, 7U, 2U, 2U }, + TensorShape{ 633U, 10U, 1U, 2U }, + TensorShape{ 781U, 5U, 2U }, + }) + { + } +}; + +/** Data set containing large softmax layer shapes. */ +class SoftmaxLayerLargeShapes final : public ShapeDataset +{ +public: + SoftmaxLayerLargeShapes() + : ShapeDataset("Shape", + { + TensorShape{ 1000U, 10U }, + TensorShape{ 3989U, 10U, 2U }, + TensorShape{ 4098U, 8U, 1U, 2U }, + TensorShape{ 7339U, 11U }, + }) + { + } +}; + } // namespace datasets } // namespace test } // namespace arm_compute |