diff options
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 = |