commit 54658b9c8cc27d513f35e078b8586767262d07f2
author Matthew Jackson <matthew.jackson@arm.com> Tue Aug 27 15:35:59 2019 +0100
committer Matteo Martincigh <matteo.martincigh@arm.com> Wed Aug 28 14:07:28 2019 +0000
tree 78af07b574caf9b9476551c894fc8f03e56df95a
parent e89ebad9cd78096d9c18a28fa01337dd622f5081

IVGCVSW-3675 Add ExecuteNetwork option to print intermediate layers

Signed-off-by: Matthew Jackson <matthew.jackson@arm.com>
Change-Id: Id7ab186ec607ff6e5ee6869c4ad562af4c40b97a

tests/InferenceModel.hpp

diff --git a/tests/InferenceModel.hpp b/tests/InferenceModel.hpp
index 13e8031..0ede898 100644
--- a/tests/InferenceModel.hpp
+++ b/tests/InferenceModel.hpp
@@ -91,6 +91,7 @@
     bool                            m_IsModelBinary;
     bool                            m_VisualizePostOptimizationModel;
     bool                            m_EnableFp16TurboMode;
+    bool                            m_PrintIntermediateLayers;
 
     Params()
         : m_ComputeDevices{}
@@ -98,6 +99,7 @@
         , m_IsModelBinary(true)
         , m_VisualizePostOptimizationModel(false)
         , m_EnableFp16TurboMode(false)
+        , m_PrintIntermediateLayers(false)
     {}
 };
 
@@ -395,6 +397,7 @@
 
             armnn::OptimizerOptions options;
             options.m_ReduceFp32ToFp16 = params.m_EnableFp16TurboMode;
+            options.m_Debug = params.m_PrintIntermediateLayers;
 
             optNet = armnn::Optimize(*network, params.m_ComputeDevices, m_Runtime->GetDeviceSpec(), options);
             if (!optNet)