diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-08-19 14:39:07 +0100 |
---|---|---|
committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-08-19 14:43:09 +0100 |
commit | a2493a0483f19fe9654be63a15badfb0834aaff6 (patch) | |
tree | a526ee366356b173947d13162ba4b34a965f23ed /include | |
parent | 37c8197c9153924ebe934d5b521c0985eab9e477 (diff) | |
download | armnn-a2493a0483f19fe9654be63a15badfb0834aaff6.tar.gz |
IVGCVSW-5012 Add importEnabled option for OptimizerOptions
* Default importEnabled to false
* Improve error messages
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I17f78986aa1d23e48b0844297a52029b1a9bbe3e
Diffstat (limited to 'include')
-rw-r--r-- | include/armnn/INetwork.hpp | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/include/armnn/INetwork.hpp b/include/armnn/INetwork.hpp index 6a143b05fb..5e8a6f2476 100644 --- a/include/armnn/INetwork.hpp +++ b/include/armnn/INetwork.hpp @@ -613,14 +613,17 @@ struct OptimizerOptions , m_Debug(false) , m_ReduceFp32ToBf16(false) , m_shapeInferenceMethod(armnn::ShapeInferenceMethod::ValidateOnly) + , m_ImportEnabled(false) {} OptimizerOptions(bool reduceFp32ToFp16, bool debug, bool reduceFp32ToBf16 = false, - ShapeInferenceMethod shapeInferenceMethod = armnn::ShapeInferenceMethod::ValidateOnly) + ShapeInferenceMethod shapeInferenceMethod = armnn::ShapeInferenceMethod::ValidateOnly, + bool importEnabled = false) : m_ReduceFp32ToFp16(reduceFp32ToFp16) , m_Debug(debug) , m_ReduceFp32ToBf16(reduceFp32ToBf16) , m_shapeInferenceMethod(shapeInferenceMethod) + , m_ImportEnabled(importEnabled) { if (m_ReduceFp32ToFp16 && m_ReduceFp32ToBf16) { @@ -639,6 +642,9 @@ struct OptimizerOptions // Infer output size when not available ShapeInferenceMethod m_shapeInferenceMethod; + + // Enable Import + bool m_ImportEnabled; }; /// Create an optimized version of the network |