aboutsummaryrefslogtreecommitdiff
path: root/python/pyarmnn/swig_generate.py
diff options
context:
space:
mode:
Diffstat (limited to 'python/pyarmnn/swig_generate.py')
-rwxr-xr-xpython/pyarmnn/swig_generate.py13
1 files changed, 7 insertions, 6 deletions
diff --git a/python/pyarmnn/swig_generate.py b/python/pyarmnn/swig_generate.py
index 7a59edaf27..72bccbf49e 100755
--- a/python/pyarmnn/swig_generate.py
+++ b/python/pyarmnn/swig_generate.py
@@ -105,12 +105,13 @@ if __name__ == "__main__":
__verbose = args.verbose
- wrap_names = ['armnn_version',
- 'armnn',
- 'armnn_caffeparser',
- 'armnn_onnxparser',
- 'armnn_tfparser',
- 'armnn_tfliteparser']
+ wrap_names = ['armnn_version',
+ 'armnn',
+ 'armnn_caffeparser',
+ 'armnn_onnxparser',
+ 'armnn_tfparser',
+ 'armnn_tfliteparser',
+ 'armnn_deserializer']
for n in wrap_names:
generate_wrap(n, f"-I{' -I'.join(armnn_includes)} ")