aboutsummaryrefslogtreecommitdiff
path: root/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i
diff options
context:
space:
mode:
authorCathal Corbett <cathal.corbett@arm.com>2021-11-18 18:17:38 +0000
committerCathal Corbett <cathal.corbett@arm.com>2021-11-22 16:01:45 +0000
commitf0836e0d7efa947f7589c476b531944078cc02d2 (patch)
treeb179d953e9ad7d44b2d2146b5e74cd48cff4e637 /python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i
parent8dd77d2093b00f195a94eb909abb526190ae6a55 (diff)
downloadarmnn-f0836e0d7efa947f7589c476b531944078cc02d2.tar.gz
IVGCVSW-6591 AddLogicalBinaryLayer to PyArmNN
* AddLogicalBinaryLayer to PyArmNN armnn_network.i * LogicalBinaryDescriptor to PyArmNN armnn_descriptors.i * Add layer to test_network_method_exists() in test_network.py * Add descriptor unit tests to test_descriptors.py Signed-off-by: Cathal Corbett <cathal.corbett@arm.com> Change-Id: I2f6288987332e1556235a9c16582e96b3a0fb641
Diffstat (limited to 'python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i')
-rw-r--r--python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i13
1 files changed, 13 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 b114edd7c4..c978e14c80 100644
--- a/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i
+++ b/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i
@@ -993,6 +993,19 @@ public:
armnn::IConnectableLayer* AddFullyConnectedLayer(const armnn::FullyConnectedDescriptor& fullyConnectedDescriptor,
const char* name = nullptr);
+ %feature("docstring",
+ "
+ Adds a LogicalBinary layer to the network.
+
+ Args:
+ logicalBinaryDescriptor (LogicalBinaryDescriptor): Description of the LogicalBinary layer.
+ name (str): Optional name for the layer.
+
+ Returns:
+ IConnectableLayer: Interface for configuring the layer.
+ ") AddLogicalBinaryLayer;
+ armnn::IConnectableLayer* AddLogicalBinaryLayer(const armnn::LogicalBinaryDescriptor& logicalBinaryDescriptor,
+ const char* name = nullptr);
};
%extend INetwork {