diff options
Diffstat (limited to 'python/pyarmnn/src')
-rw-r--r-- | python/pyarmnn/src/pyarmnn/__init__.py | 2 | ||||
-rw-r--r-- | python/pyarmnn/src/pyarmnn/swig/modules/armnn_descriptors.i | 19 | ||||
-rw-r--r-- | python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i | 13 |
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 { |