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

Makes sure tf.print adds end when printing interactively

Browse Source 
PiperOrigin-RevId: 246176330
This commit is contained in:
A. Unique TensorFlower 2019-05-01 12:20:55 -07:00 committed by TensorFlower Gardener
parent ac6e979676
commit 014d9bcd23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/kernels/logging_ops.cc
View File

@ -183,7 +183,7 @@ class PrintV2Op : public OpKernel {
auto listeners = logging::GetListeners(); auto listeners = logging::GetListeners();
if (!listeners->empty()) { if (!listeners->empty()) {
for (auto& listener : *listeners) { for (auto& listener : *listeners) {
listener(msg.c_str()); listener(ended_msg.c_str());
} }
} else if (output_stream_ == "stdout") { } else if (output_stream_ == "stdout") {
std::cout << ended_msg << std::flush; std::cout << ended_msg << std::flush;