diff options
Diffstat (limited to 'python/pyarmnn/src')
-rw-r--r-- | python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i | 42 |
1 files changed, 42 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 0b7f55d1cc..bfaa55919b 100644 --- a/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i +++ b/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i @@ -597,6 +597,20 @@ public: %feature("docstring", + " + Adds a Broadcast_To layer to the network. + + Args: + broadcastToDescriptor (BroadcastToDescriptor): Descriptor for the explicit broadcat operation. + name (str): Optional name for the layer. + + Returns: + IConnectableLayer: Interface for configuring the layer. + ") AddBroadcastToLayer; + armnn::IConnectableLayer* AddBroadcastToLayer(const armnn::BroadcastToDescriptor& broadcastToDescriptor, + const char* name = nullptr); + + %feature("docstring", " Adds a Division layer to the network. @@ -1148,6 +1162,20 @@ public: const char* name = nullptr); %feature("docstring", + " + Adds a Fused layer to the network. This is a precompiled layer for fused operator that merges Add + Mul + Add. + + Args: + fusedDescriptor (FusedDescriptor): Description of the fused layer. + name (str): Optional name for the layer. + + Returns: + IConnectableLayer: Interface for configuring the layer. + ") AddFusedLayer; + armnn::IConnectableLayer* AddFusedLayer(const armnn::FusedDescriptor& fusedDescriptor, + const char* name = nullptr); + + %feature("docstring", " Adds a LogicalBinary layer to the network. @@ -1162,6 +1190,20 @@ public: const char* name = nullptr); %feature("docstring", + " + Adds a ScatterND layer to the network. + + Args: + scatterndDescriptor (ScatterNdDescriptor): Description of the tile layer. + name (str): Optional name for the layer. + + Returns: + IConnectableLayer: Interface for configuring the layer. + ") AddScatterNdLayer; + armnn::IConnectableLayer* AddScatterNdLayer(const armnn::ScatterNdDescriptor& scatterndDescriptor, + const char* name = nullptr); + + %feature("docstring", " Adds a Tile layer to the network. |