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

[tf.data] Add traceme for prefetch with slack time

Browse Source 
PiperOrigin-RevId: 256402325
This commit is contained in:
Rachel Lim 2019-07-03 11:09:23 -07:00 committed by TensorFlower Gardener
parent 9f9b8a5101
commit 0cd2977c08
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/core/kernels/data/prefetch_dataset_op.cc
View File

@ -128,7 +128,13 @@ class PrefetchDatasetOp::Dataset : public DatasetBase {
tf_shared_lock l(mu_);
buffer_limit = auto_tuner_.buffer_limit();
}
return strings::StrCat(prefix(), "#buffer_limit=", buffer_limit, "#");
string prefetch_with_slack_trace = "";
if (dataset()->slack_period_ > 0) {
int64 slack_us = slack_us_;
prefetch_with_slack_trace = strings::StrCat(",slack=", slack_us);
}
return strings::StrCat(prefix(), "#buffer_limit=", buffer_limit,
prefetch_with_slack_trace, "#");
}
Status Initialize(IteratorContext* ctx) override {