From aeee9adbbee8c60fdfecc5bcc81b18e0748e4ee1 Mon Sep 17 00:00:00 2001 From: Ryan OShea Date: Thu, 18 Nov 2021 17:43:55 +0000 Subject: IVGCVSW-6587 AddShapeLayer to PyArmNN * Add layer to armnn.network.i * Add layer to test_network.py Signed-off-by: Ryan OShea Change-Id: I7554a0ff62d81fcc99aa68204ae5f1cc3965c44b --- python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i | 12 ++++++++++++ python/pyarmnn/test/test_network.py | 1 + 2 files changed, 13 insertions(+) (limited to 'python') 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 @@ -840,6 +840,18 @@ public: const char* name = nullptr); + %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. diff --git a/python/pyarmnn/test/test_network.py b/python/pyarmnn/test/test_network.py index ff1c66edc4..f07f424d52 100644 --- a/python/pyarmnn/test/test_network.py +++ b/python/pyarmnn/test/test_network.py @@ -228,6 +228,7 @@ def test_serialize_to_dot_mode_readonly(network_file, get_runtime, tmpdir): 'AddRankLayer', 'AddReshapeLayer', 'AddResizeLayer', + 'AddShapeLayer', 'AddSliceLayer', 'AddSoftmaxLayer', 'AddSpaceToBatchNdLayer', -- cgit v1.2.1