diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-08-21 10:03:49 +0100 |
---|---|---|
committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-08-21 10:05:11 +0100 |
commit | ea063dfe1533162581ee968ae44189edd3f78abc (patch) | |
tree | 4a2fff825792ddcee60716bedd33fb6ae82e3931 /python/pyarmnn/src | |
parent | 0fe7376f6982842e32591f3996d5ee650d22156f (diff) | |
download | armnn-ea063dfe1533162581ee968ae44189edd3f78abc.tar.gz |
IVGCVSW-5200 Add import enabled optimizer options to PyArmNN
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: Ic7c288fd829d7f1f1ae2910c47fbccdd635def8f
Diffstat (limited to 'python/pyarmnn/src')
-rw-r--r-- | python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i b/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i index 4665e6087e..57bf355253 100644 --- a/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i +++ b/python/pyarmnn/src/pyarmnn/swig/modules/armnn_network.i @@ -24,6 +24,7 @@ Contains: that can not be reduced will be left in Fp32. m_ReduceFp32ToFp16 (bool): Reduces Fp32 network to Fp16 for faster processing. Layers that can not be reduced will be left in Fp32. + m_ImportEnabled (bool): Enable memory import. ") OptimizerOptions; struct OptimizerOptions @@ -32,11 +33,13 @@ struct OptimizerOptions OptimizerOptions(bool reduceFp32ToFp16, bool debug, - bool reduceFp32ToBf16 = false); + bool reduceFp32ToBf16 = false, + bool importEnabled = false); bool m_ReduceFp32ToBf16; bool m_ReduceFp32ToFp16; bool m_Debug; + bool m_ImportEnabled; }; %feature("docstring", |