From d6afedc775220f17317f1835a4d18b72a54525de Mon Sep 17 00:00:00 2001 From: Chunosov Date: Mon, 6 Nov 2017 22:09:45 +0700 Subject: COMPMID-661: softmax-fp32 optimisation (#14) Change-Id: I2007af1ed9dcf68065cf412aa50f73a2025b31a6 Reviewed-on: http://mpd-gerrit.cambridge.arm.com/94605 Reviewed-by: Gian Marco Iodice Tested-by: Kaizen --- tests/datasets/ShapeDatasets.h | 34 ++++++++++++++++++++++++++++++++++ 1 file changed, 34 insertions(+) (limited to 'tests/datasets/ShapeDatasets.h') 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 -- cgit v1.2.1