From 738c2e6a647b886750e1bc3daa6dd615a0244baa Mon Sep 17 00:00:00 2001
From: keidav01 <keith.davis@arm.com>
Date: Tue, 11 Dec 2018 16:14:20 +0000
Subject: IVGCVSW-1434 Add debug mode to Optimizer     * Modified optimizer to
 support debug mode via DebugLayer

Change-Id: Ic8f313778e55540c182cf99876c44a0823be04c6
---
 include/armnn/INetwork.hpp | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

(limited to 'include')

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
-- 
cgit v1.2.1