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>

1 files changed, 22 insertions, 3 deletions

diff --git a/tests/datasets/ShapeDatasets.h b/tests/datasets/ShapeDatasets.h
index ddcf6a7a95..aa653ace60 100644
--- a/tests/datasets/ShapeDatasets.h
+++ b/tests/datasets/ShapeDatasets.h
@@ -101,14 +101,14 @@ public:
         : 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 @@ public:
     }
 };
 
+/** 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