diff options
Diffstat (limited to 'python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i')
| -rw-r--r-- | python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i | 55 |
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: |