diff options
author | Jan Eilers <jan.eilers@arm.com> | 2021-09-07 12:46:15 +0100 |
---|---|---|
committer | Jan Eilers <jan.eilers@arm.com> | 2021-09-13 10:05:00 +0100 |
commit | 284b5d19c8cf446b7eec16ea560c544ac39008c1 (patch) | |
tree | 8a4d34db3fff091bacd15b004b9c6e6e4b8248ba /tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp | |
parent | 65d5d2ddf77c69e76643e40440aa986defe6d0d7 (diff) | |
download | armnn-284b5d19c8cf446b7eec16ea560c544ac39008c1.tar.gz |
Add 'do-not-print-output' option to ExNet
Signed-off-by: Jan Eilers <jan.eilers@arm.com>
Change-Id: I10fb010ee8d3f813d2264cefb526f352e30d7046
Diffstat (limited to 'tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp')
-rw-r--r-- | tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp b/tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp index d92c17c5e5..0abda4f8ee 100644 --- a/tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp +++ b/tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp @@ -26,7 +26,8 @@ struct TensorPrinter TensorPrinter(const std::string& binding, const armnn::TensorInfo& info, const std::string& outputTensorFile, - bool dequantizeOutput); + bool dequantizeOutput, + bool printToConsole = true); void operator()(const std::vector<float>& values); @@ -48,6 +49,7 @@ private: int m_Offset; std::string m_OutputTensorFile; bool m_DequantizeOutput; + bool m_PrintToConsole; }; using TContainer = |