1 files changed, 14 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.