diff options
author | Ryan OShea <ryan.oshea3@arm.com> | 2021-11-18 17:43:55 +0000 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2021-11-23 10:39:05 +0000 |
commit | aeee9adbbee8c60fdfecc5bcc81b18e0748e4ee1 (patch) | |
tree | 37e8a2a2ebd98ef90a256f3e2da6e75a05cbc0c2 /python/pyarmnn/src | |
parent | 2b4182fa86c4efe43f3a631266b5185d8a725fa1 (diff) | |
download | armnn-aeee9adbbee8c60fdfecc5bcc81b18e0748e4ee1.tar.gz |
IVGCVSW-6587 AddShapeLayer to PyArmNN
* Add layer to armnn.network.i
* Add layer to test_network.py
Signed-off-by: Ryan OShea <ryan.oshea3@arm.com>
Change-Id: I7554a0ff62d81fcc99aa68204ae5f1cc3965c44b
Diffstat (limited to 'python/pyarmnn/src')
-rw-r--r-- | python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i | 12 |
1 files changed, 12 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 f1cae5605b..0dc016fb96 100644 --- a/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i +++ b/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i @@ -842,6 +842,18 @@ public: %feature("docstring", " + Adds a Shape layer to the network. + + Args: + name(str): Optional name for the layer. + + Returns: + IConnectableLayer: Interface for configuring the layer + ") AddShapeLayer; + armnn::IConnectableLayer* AddShapeLayer(const char* name = nullptr); + + %feature("docstring", + " Adds a Slice layer to the network. Args: |