diff options
author | Matthew Jackson <matthew.jackson@arm.com> | 2019-08-27 15:35:59 +0100 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-08-28 14:07:28 +0000 |
commit | 54658b9c8cc27d513f35e078b8586767262d07f2 (patch) | |
tree | 78af07b574caf9b9476551c894fc8f03e56df95a /tests/InferenceModel.hpp | |
parent | e89ebad9cd78096d9c18a28fa01337dd622f5081 (diff) | |
download | armnn-54658b9c8cc27d513f35e078b8586767262d07f2.tar.gz |
IVGCVSW-3675 Add ExecuteNetwork option to print intermediate layers
Signed-off-by: Matthew Jackson <matthew.jackson@arm.com>
Change-Id: Id7ab186ec607ff6e5ee6869c4ad562af4c40b97a
Diffstat (limited to 'tests/InferenceModel.hpp')
-rw-r--r-- | tests/InferenceModel.hpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tests/InferenceModel.hpp b/tests/InferenceModel.hpp index 13e80319f4..0ede8983cf 100644 --- a/tests/InferenceModel.hpp +++ b/tests/InferenceModel.hpp @@ -91,6 +91,7 @@ struct Params bool m_IsModelBinary; bool m_VisualizePostOptimizationModel; bool m_EnableFp16TurboMode; + bool m_PrintIntermediateLayers; Params() : m_ComputeDevices{} @@ -98,6 +99,7 @@ struct Params , m_IsModelBinary(true) , m_VisualizePostOptimizationModel(false) , m_EnableFp16TurboMode(false) + , m_PrintIntermediateLayers(false) {} }; @@ -395,6 +397,7 @@ public: 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) |