COMPMID-522 - Added support for GlobalPooling in CLPoolingLayer and CLFlattening for 3D tensor
Change-Id: Ifc7db1e4d4af322a4dcbfeb3e132e5c326596872
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/86618
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
diff --git a/tests/datasets/ShapeDatasets.h b/tests/datasets/ShapeDatasets.h
index ddcf6a7..aa653ac 100644
--- a/tests/datasets/ShapeDatasets.h
+++ b/tests/datasets/ShapeDatasets.h
@@ -101,14 +101,14 @@
: ShapeDataset("Shape",
{
// Batch size 1
- TensorShape{ 7U, 7U },
+ TensorShape{ 9U, 9U },
TensorShape{ 27U, 13U, 2U },
TensorShape{ 128U, 64U, 1U, 3U },
// Batch size 4
- TensorShape{ 7U, 7U, 3U, 4U },
+ TensorShape{ 9U, 9U, 3U, 4U },
TensorShape{ 27U, 13U, 2U, 4U },
// Arbitrary batch size
- TensorShape{ 7U, 7U, 3U, 5U }
+ TensorShape{ 9U, 9U, 3U, 5U }
})
{
}
@@ -218,6 +218,25 @@
}
};
+/** Data set containing global pooling tensor shapes. */
+class GlobalPoolingShapes final : public ShapeDataset
+{
+public:
+ GlobalPoolingShapes()
+ : ShapeDataset("Shape",
+ {
+ // Batch size 1
+ TensorShape{ 9U, 9U },
+ TensorShape{ 13U, 13U, 2U },
+ TensorShape{ 27U, 27U, 1U, 3U },
+ // Batch size 4
+ TensorShape{ 31U, 31U, 3U, 4U },
+ TensorShape{ 34U, 34U, 2U, 4U }
+ })
+ {
+ }
+};
+
} // namespace datasets
} // namespace test
} // namespace arm_compute