IVGCVSW-6749 Add Pooling3d to PyArmnn

 * Add layer to __init__.py
 * Add descriptor for Pooling3d
 * Add descriptor test for Pooling3d
 * Add network test for Pooling3d layer

Signed-off-by: Ryan OShea <ryan.oshea3@arm.com>
Change-Id: Id5e1587a89d3ffb5bee7764a92b299fa43a2ae35

1 files changed, 15 insertions, 1 deletions

diff --git a/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i b/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i
index eb8dede260..1957288003 100644
--- a/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i
+++ b/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i
@@ -803,7 +803,21 @@ public:
             IConnectableLayer: Interface for configuring the layer.
         ") AddPooling2dLayer;
     armnn::IConnectableLayer* AddPooling2dLayer(const armnn::Pooling2dDescriptor& pooling2dDescriptor,
-        const char* name = nullptr);
+                                                const char* name = nullptr);
+
+    %feature("docstring",
+        "
+        Adds a 3D Pooling layer to the network. Type of 3D pooling is decided by the configuration.
+
+        Args:
+            pooling3dDescriptor (Pooling3dDescriptor): Configuration for the 3D pooling layer.
+            name (str): Optional name for the layer.
+
+        Returns:
+            IConnectableLayer: Interface for configuring the layer.
+        ") AddPooling3dLayer;
+    armnn::IConnectableLayer* AddPooling3dLayer(const armnn::Pooling3dDescriptor& pooling3dDescriptor,
+                                                const char* name = nullptr);
 
     %feature("docstring",
         "