1 files changed, 41 insertions, 0 deletions
diff --git a/python/pyarmnn/src/pyarmnn/swig/typemaps/network_optimize.i b/python/pyarmnn/src/pyarmnn/swig/typemaps/network_optimize.i
new file mode 100644
index 0000000000..05df82bdd1
--- /dev/null
+++ b/python/pyarmnn/src/pyarmnn/swig/typemaps/network_optimize.i
@@ -0,0 +1,41 @@
+//
+// Copyright © 2017 Arm Ltd. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+%define %optimize_typemap_out
+    %typemap(out) (std::pair<armnn::IOptimizedNetwork*, std::vector<std::string>>) {
+        PyObject * network = SWIG_NewPointerObj(SWIG_as_voidptr($1.first), SWIGTYPE_p_armnn__IOptimizedNetwork, SWIG_POINTER_OWN);
+        $result = PyTuple_New(2);
+
+        // Convert vector to fixed-size tuple
+        std::vector<std::string> strings = $1.second;
+        Py_ssize_t size = strings.size();
+
+        // New reference. Need to Py_DECREF
+        PyObject* errMsgTuple = PyTuple_New(size);
+
+        if (!errMsgTuple) {
+            Py_XDECREF(errMsgTuple);
+            return PyErr_NoMemory();
+        }
+
+        for (Py_ssize_t i = 0; i < size; i++) {
+            // New reference. Need to Py_DECREF
+            PyObject *string = PyString_FromString(strings[i].c_str());
+
+            if (!string) {
+                Py_XDECREF(string);
+                return PyErr_NoMemory();
+            }
+            PyTuple_SetItem(errMsgTuple, i, string);
+        }
+
+        // Create result tuple
+        PyTuple_SetItem($result, 0, network);
+        PyTuple_SetItem($result, 1, errMsgTuple);
+    }
+%enddef
+
+%define %clear_optimize_typemap_out
+    %typemap(out) (std::pair<armnn::IOptimizedNetwork*, std::vector<std::string>>)
+%enddef