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 ++++++++++++++ 1 file changed, 14 insertions(+) (limited to 'python/pyarmnn/src') 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. -- cgit v1.2.1