[tf.data] Add vlogging for entering/exiting GetNext ops.
This is helpful for identifying whether a process is stuck in tf.data.
PiperOrigin-RevId: 427799503
Change-Id: I2d136a421c4815408e56cccb1ef7f1deda90c28f
diff --git a/tensorflow/core/kernels/data/iterator_ops.cc b/tensorflow/core/kernels/data/iterator_ops.cc
index 5bb26d7..9f84d20 100644
--- a/tensorflow/core/kernels/data/iterator_ops.cc
+++ b/tensorflow/core/kernels/data/iterator_ops.cc
@@ -920,6 +920,10 @@
}
Status IteratorGetNextOp::DoCompute(OpKernelContext* ctx) {
+ VLOG(3) << "IteratorGetNextOp enter. iter_id=" << ctx->frame_iter().iter_id;
+ auto cleanup = gtl::MakeCleanup([ctx] {
+ VLOG(3) << "IteratorGetNextOp exit. iter_id=" << ctx->frame_iter().iter_id;
+ });
profiler::TraceMe traceme(
[&] {
return profiler::TraceMeEncode(
@@ -949,6 +953,12 @@
}
Status IteratorGetNextAsOptionalOp::DoCompute(OpKernelContext* ctx) {
+ VLOG(3) << "IteratorGetNextAsOptionalOp exit. iter_id="
+ << ctx->frame_iter().iter_id;
+ auto cleanup = gtl::MakeCleanup([ctx] {
+ VLOG(3) << "IteratorGetNextAsOptionalOp exit. iter_id="
+ << ctx->frame_iter().iter_id;
+ });
profiler::TraceMe traceme(
[&] {
return profiler::TraceMeEncode(
