diff options
| author | keidav01 <keith.davis@arm.com> | 2018-12-11 16:14:20 +0000 |
|---|---|---|
| committer | Keith Davis Arm <keith.davis@arm.com> | 2018-12-11 17:11:20 +0000 |
| commit | 738c2e6a647b886750e1bc3daa6dd615a0244baa (patch) | |
| tree | ef88c6a5ee1076471bd36b85c30afb36a68bbfc9 /include | |
| parent | 44a7167e0f13dc1d703cd573f57636fde711c618 (diff) | |
| download | armnn-738c2e6a647b886750e1bc3daa6dd615a0244baa.tar.gz | |
IVGCVSW-1434 Add debug mode to Optimizer
* Modified optimizer to support debug mode via DebugLayer
Change-Id: Ic8f313778e55540c182cf99876c44a0823be04c6
Diffstat (limited to 'include')
| -rw-r--r-- | include/armnn/INetwork.hpp | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/include/armnn/INetwork.hpp b/include/armnn/INetwork.hpp index 31f1dc2b91..33181ce138 100644 --- a/include/armnn/INetwork.hpp +++ b/include/armnn/INetwork.hpp @@ -349,15 +349,22 @@ protected: struct OptimizerOptions { - OptimizerOptions() : m_ReduceFp32ToFp16(false) {} + OptimizerOptions() : + m_ReduceFp32ToFp16(false) + , m_Debug(false) + {} - OptimizerOptions(bool reduceFp32ToFp16) + OptimizerOptions(bool reduceFp32ToFp16, bool debug) : m_ReduceFp32ToFp16(reduceFp32ToFp16) + , m_Debug(debug) { } // Reduce Fp32 data to Fp16 for faster processing bool m_ReduceFp32ToFp16; + + // Add debug data for easier troubleshooting + bool m_Debug; }; /// Create an optimized version of the network |