IVGCVSW-7703 Ensure PyArmNN has been updated with new features added in ArmNN

 * Adds BatchMatMul layer and descriptor to pyarmnn
 * Adds ReverseV2 layer to pyarmnn
 * Adds ElementWiseBinary layer and descriptor to pyarmnn
 * Adds Tile layer and descriptor to pyarmnn
 * Adds network test for each layer

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

1 files changed, 55 insertions, 0 deletions

diff --git a/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i b/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i
index c9eef8630d..0b7f55d1cc 100644
--- a/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i
+++ b/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i
@@ -396,6 +396,21 @@ public:
 
     %feature("docstring",
         "
+        Adds a Batch Matrix Multiplication layer to the network.
+
+        Args:
+            desc (BatchMatMulDescriptor): Parameters for the BatchMatMul layer.
+            name (str): Optional name for the layer.
+
+        Returns:
+            IConnectableLayer: Interface for configuring the layer.
+        ") AddBatchMatMulLayer;
+    armnn::IConnectableLayer* AddBatchMatMulLayer(const armnn::BatchMatMulDescriptor& desc,
+        const char* name = nullptr);
+
+
+    %feature("docstring",
+        "
         Adds a Batch Normalization layer to the network.
 
         Args:
@@ -595,6 +610,20 @@ public:
 
     %feature("docstring",
         "
+        Adds an Elementwise Binary layer to the network. Type of binary operation to use is decided by elementwiseBinaryDescriptor. Binary operations supported are (Add, Div, Maximum, Minimum, Mul, Sub, SqDiff, Power)
+
+        Args:
+            elementwiseBinaryDescriptor (ElementwiseBinaryDescriptor): ElementwiseBinaryDescriptor to configure the choice of binary operation added to the network.
+            name (str): Optional name for the layer.
+
+        Returns:
+            IConnectableLayer: Interface for configuring the layer.
+        ") AddElementwiseBinaryLayer;
+    armnn::IConnectableLayer* AddElementwiseBinaryLayer(const ElementwiseBinaryDescriptor& elementwiseBinaryDescriptor,
+                                                       const char* name = nullptr);
+
+    %feature("docstring",
+        "
         Adds an Elementwise Unary layer to the network. Type of unary operation to use is decided by elementwiseUnaryDescriptor. Unary operations supported are (Abs, Exp, Neg, Rsqrt, Sqrt)
 
         Args:
@@ -943,6 +972,18 @@ public:
 
     %feature("docstring",
         "
+        Adds a ReverseV2 layer to the network.
+
+        Args:
+            name (str): Optional name for the layer.
+
+        Returns:
+            IConnectableLayer: Interface for configuring the layer.
+        ") AddReverseV2Layer;
+    armnn::IConnectableLayer* AddReverseV2Layer(const char* name = nullptr);
+
+    %feature("docstring",
+        "
         Adds a Shape layer to the network.
 
         Args:
@@ -1122,6 +1163,20 @@ public:
 
     %feature("docstring",
         "
+        Adds a Tile layer to the network.
+
+        Args:
+            tileDescriptor (TileDescriptor): Description of the tile layer.
+            name (str): Optional name for the layer.
+
+        Returns:
+            IConnectableLayer: Interface for configuring the layer.
+        ") AddTileLayer;
+    armnn::IConnectableLayer* AddTileLayer(const armnn::TileDescriptor& tileDescriptor,
+                                                const char* name = nullptr);
+
+    %feature("docstring",
+        "
         Adds a Transpose layer to the network.
 
         Args: