diff options
author | Keith Davis <keith.davis@arm.com> | 2021-08-09 16:49:18 +0100 |
---|---|---|
committer | Matthew Sloyan <matthew.sloyan@arm.com> | 2021-08-10 14:07:27 +0000 |
commit | f487486c843a38fced90229923433d09f99fc2e5 (patch) | |
tree | 63a2268196279a5c97c3b9dea86024f0bf4a751e /tests/InferenceModel.hpp | |
parent | bcd860a30eba22bb2ba0943ad705734ce0ec5b23 (diff) | |
download | armnn-f487486c843a38fced90229923433d09f99fc2e5.tar.gz |
IVGCVSW-6292 Allow profiling details to be switched off during profiling
* Add switch for network details during profiling
Signed-off-by: Keith Davis <keith.davis@arm.com>
Change-Id: I8bd49fd58f0e0255598106e9ab36806ee78391d6
Diffstat (limited to 'tests/InferenceModel.hpp')
-rw-r--r-- | tests/InferenceModel.hpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/tests/InferenceModel.hpp b/tests/InferenceModel.hpp index 4d2b167522..1db287f453 100644 --- a/tests/InferenceModel.hpp +++ b/tests/InferenceModel.hpp @@ -100,6 +100,7 @@ struct Params bool m_InferOutputShape; bool m_EnableFastMath; bool m_SaveCachedNetwork; + bool m_OutputDetailsToStdOut; std::string m_CachedNetworkFilePath; unsigned int m_NumberOfThreads; std::string m_MLGOTuningFilePath; @@ -119,6 +120,7 @@ struct Params , m_InferOutputShape(false) , m_EnableFastMath(false) , m_SaveCachedNetwork(false) + , m_OutputDetailsToStdOut(false) , m_CachedNetworkFilePath("") , m_NumberOfThreads(0) , m_MLGOTuningFilePath("") @@ -489,7 +491,8 @@ public: armnn::INetworkProperties networkProperties(params.m_AsyncEnabled, armnn::MemorySource::Undefined, armnn::MemorySource::Undefined, - enableProfiling); + enableProfiling, + params.m_OutputDetailsToStdOut); std::string errorMessage; ret = m_Runtime->LoadNetwork(m_NetworkIdentifier, std::move(optNet), errorMessage, networkProperties); |