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>
diff --git a/tests/datasets/ShapeDatasets.h b/tests/datasets/ShapeDatasets.h
index 86ed2b2..45f5d1c 100644
--- a/tests/datasets/ShapeDatasets.h
+++ b/tests/datasets/ShapeDatasets.h
@@ -269,6 +269,40 @@
}
};
+/** 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