experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

This allows us to grep & process the output easier.

Browse Source 
PiperOrigin-RevId: 260989162
This commit is contained in:
A. Unique TensorFlower 2019-07-31 13:38:40 -07:00 committed by TensorFlower Gardener
parent 75877ac1ea
commit 6c8f1ee32a
1 changed files with 13 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tensorflow/lite/tools/benchmark/logging.h
View File

@ -46,10 +46,19 @@ class LoggingWrapper {
std::stringstream& Stream() { return stream_; } std::stringstream& Stream() { return stream_; }
~LoggingWrapper() { ~LoggingWrapper() {
if (should_log_) { if (should_log_) {
switch (severity_) {
case LogSeverity::INFO:
case LogSeverity::WARN:
std::cout << stream_.str() << std::endl;
break;
case LogSeverity::ERROR:
std::cerr << stream_.str() << std::endl;
break;
case LogSeverity::FATAL:
std::cerr << stream_.str() << std::endl; std::cerr << stream_.str() << std::endl;
if (severity_ == LogSeverity::FATAL) {
std::flush(std::cerr); std::flush(std::cerr);
std::abort(); std::abort();
break;
} }
} }
} }