Makes sure tf.print adds end when printing interactively

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

View File

@ -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;