aboutsummaryrefslogtreecommitdiff
path: root/python/pyarmnn/src
diff options
context:
space:
mode:
Diffstat (limited to 'python/pyarmnn/src')
-rw-r--r--python/pyarmnn/src/pyarmnn/__init__.py2
-rw-r--r--python/pyarmnn/src/pyarmnn/swig/modules/armnn_descriptors.i19
-rw-r--r--python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i13
3 files changed, 34 insertions, 0 deletions
diff --git a/python/pyarmnn/src/pyarmnn/__init__.py b/python/pyarmnn/src/pyarmnn/__init__.py
index 44992522c0..5f3e5ff521 100644
--- a/python/pyarmnn/src/pyarmnn/__init__.py
+++ b/python/pyarmnn/src/pyarmnn/__init__.py
@@ -84,6 +84,8 @@ from ._generated.pyarmnn import ComparisonDescriptor, ComparisonOperation_Equal,
ComparisonOperation_LessOrEqual, ComparisonOperation_NotEqual
from ._generated.pyarmnn import UnaryOperation_Abs, UnaryOperation_Exp, UnaryOperation_Sqrt, UnaryOperation_Rsqrt, \
UnaryOperation_Neg, ElementwiseUnaryDescriptor
+from ._generated.pyarmnn import LogicalBinaryOperation_LogicalAnd, LogicalBinaryOperation_LogicalOr, \
+ LogicalBinaryDescriptor
from ._generated.pyarmnn import Convolution2dDescriptor, DepthToSpaceDescriptor, DepthwiseConvolution2dDescriptor, \
DetectionPostProcessDescriptor, FakeQuantizationDescriptor, FillDescriptor, FullyConnectedDescriptor, \
GatherDescriptor, InstanceNormalizationDescriptor, LstmDescriptor, L2NormalizationDescriptor, MeanDescriptor
diff --git a/python/pyarmnn/src/pyarmnn/swig/modules/armnn_descriptors.i b/python/pyarmnn/src/pyarmnn/swig/modules/armnn_descriptors.i
index a050722bb9..8844cea81a 100644
--- a/python/pyarmnn/src/pyarmnn/swig/modules/armnn_descriptors.i
+++ b/python/pyarmnn/src/pyarmnn/swig/modules/armnn_descriptors.i
@@ -1036,6 +1036,25 @@ struct TransposeConvolution2dDescriptor
bool operator ==(const TransposeConvolution2dDescriptor& rhs) const;
};
+%feature("docstring",
+ "
+ A descriptor for the LogicalBinary layer. See `INetwork.AddLogicalBinaryLayer()`.
+
+ Contains:
+ m_Operation (int): Specifies the logical operation to execute.
+ (0: `LogicalBinaryOperation_LogicalAnd`, 1: `LogicalBinaryOperation_LogicalOr`)
+ Default: 0: `LogicalBinaryOperation_LogicalAnd`.
+
+ ") LogicalBinaryDescriptor;
+struct LogicalBinaryDescriptor
+{
+ LogicalBinaryDescriptor();
+ LogicalBinaryDescriptor(LogicalBinaryOperation operation);
+
+ LogicalBinaryOperation m_Operation;
+
+ bool operator ==(const LogicalBinaryDescriptor &rhs) const;
+};
using ConcatDescriptor = OriginsDescriptor;
using LogSoftmaxDescriptor = SoftmaxDescriptor;
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 {