aboutsummaryrefslogtreecommitdiff
path: root/python/pyarmnn/swig_generate.py
diff options
context:
space:
mode:
authorwangg <gemini910621@gmail.com>2020-08-26 01:44:32 +0000
committerJim Flynn <jim.flynn@arm.com>2020-10-21 12:16:32 +0000
commit5f960d92dbefabf708e8f299789b35f6cdf2d919 (patch)
tree7309f46bfdef2a319c987d0492888d4aee93a509 /python/pyarmnn/swig_generate.py
parent45274909b06a4882ada92899c58ee66194446135 (diff)
downloadarmnn-5f960d92dbefabf708e8f299789b35f6cdf2d919.tar.gz
Add IDeserializer support in pyarmnn
Resources required for new unit tests are included for review. Signed-off-by: Guanqun Wang gemini910621@gmail.com Change-Id: Iead6cb5beaf824a6f467ad9da4aede5719ebe4ec
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)} ")