diff options
-rw-r--r-- | python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i | 14 | ||||
-rw-r--r-- | python/pyarmnn/test/test_network.py | 1 |
2 files changed, 15 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 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', |