aboutsummaryrefslogtreecommitdiff
path: root/python/pyarmnn/src
diff options
context:
space:
mode:
Diffstat (limited to 'python/pyarmnn/src')
-rw-r--r--python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i42
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.