Makes sure tf.print adds end when printing interactively
PiperOrigin-RevId: 246176330
This commit is contained in:
parent
ac6e979676
commit
014d9bcd23
@ -183,7 +183,7 @@ class PrintV2Op : public OpKernel {
|
||||
auto listeners = logging::GetListeners();
|
||||
if (!listeners->empty()) {
|
||||
for (auto& listener : *listeners) {
|
||||
listener(msg.c_str());
|
||||
listener(ended_msg.c_str());
|
||||
}
|
||||
} else if (output_stream_ == "stdout") {
|
||||
std::cout << ended_msg << std::flush;
|
||||
|
Loading…
Reference in New Issue
Block a user