diff options
author | Ryan OShea <ryan.oshea3@arm.com> | 2021-11-18 16:52:41 +0000 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2021-11-26 10:27:39 +0000 |
commit | 09a052251b3c4c14c093e79d83c36504746890be (patch) | |
tree | db70f9e26250cf220db44265ec159878e53a1aa6 /python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i | |
parent | e0d1fc5ad4f47bfdff2fe2f3fbcebf68669257dd (diff) | |
download | armnn-09a052251b3c4c14c093e79d83c36504746890be.tar.gz |
IVGCVSW-6589 AddReduceLayer to PyArmNN
* AddReduceLayer to PyArmNN armnn_network.i
* AddReduceDescriptor to PyArmNN armnn_descriptors.i
* Add layer to test_network.py
* Add descriptor unit tests to test_descriptors.py
Signed-off-by: Ryan OShea <ryan.oshea3@arm.com>
Change-Id: I2a8706052baf8f9c0038a5ea327d73b8f3ea1961
Diffstat (limited to 'python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i')
-rw-r--r-- | python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i | 13 |
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 789e428e46..643f99f53a 100644 --- a/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i +++ b/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i @@ -856,6 +856,19 @@ public: ") AddRankLayer; armnn::IConnectableLayer* AddRankLayer(const char* name = nullptr); + %feature("docstring", + " + Adds a Reduce layer to the network. + + Args: + reduceDescriptor (ReduceDescriptor): Parameters for the reduce operation. + name (str): Optional name for the layer. + + Returns: + IConnectableLayer: Interface for configuring the layer. + ") AddReduceLayer; + armnn::IConnectableLayer* AddReduceLayer(const armnn::ReduceDescriptor& reduceDescriptor, + const char* name = nullptr); %feature("docstring", " |