From 87dc8122b97f5e3b7f81b7434097b9be2db9c902 Mon Sep 17 00:00:00 2001 From: Teresa Charlin Date: Mon, 22 Nov 2021 15:34:26 +0000 Subject: IVGCVSW-6588 AddCastLayer to PyArmNN Signed-off-by: Teresa Charlin Change-Id: If9c6902396b9f3cec3b38b70ccfa12fcbb3fd5a7 --- python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i | 14 ++++++++++++++ python/pyarmnn/test/test_network.py | 1 + 2 files changed, 15 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 0dc016fb96..3e3a36570e 100644 --- a/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i +++ b/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i @@ -427,6 +427,20 @@ public: armnn::IConnectableLayer* AddBatchToSpaceNdLayer(const armnn::BatchToSpaceNdDescriptor& batchToSpaceNdDescriptor, const char* name = nullptr); + + %feature("docstring", + " + Adds a Cast layer to the network. + + Args: + name (str): Optional name for the layer. + + Returns: + IConnectableLayer: Interface for configuring the layer. + ") AddCastLayer; + armnn::IConnectableLayer* AddCastLayer(const char* name = nullptr); + + %feature("docstring", " Adds a Comparison layer to the network. diff --git a/python/pyarmnn/test/test_network.py b/python/pyarmnn/test/test_network.py index f07f424d52..d734f619f7 100644 --- a/python/pyarmnn/test/test_network.py +++ b/python/pyarmnn/test/test_network.py @@ -192,6 +192,7 @@ def test_serialize_to_dot_mode_readonly(network_file, get_runtime, tmpdir): 'AddArgMinMaxLayer', 'AddBatchNormalizationLayer', 'AddBatchToSpaceNdLayer', + 'AddCastLayer', 'AddComparisonLayer', 'AddConcatLayer', 'AddConstantLayer', -- cgit v1.2.1