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

[tf.data] Reducing tracing overhead.

Browse Source 
PiperOrigin-RevId: 249518292
This commit is contained in:
Jiri Simsa 2019-05-22 14:17:32 -07:00 committed by TensorFlower Gardener
parent 8877e397f6
commit e5678b1560
3 changed files with 9 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/core/kernels/data/experimental/map_and_batch_dataset_op.cc
View File

@ -208,11 +208,9 @@ class MapAndBatchDatasetOp : public UnaryDatasetOpKernel {
}
string BuildTraceMeName() override {
int64 parallelism;
{
tf_shared_lock l(*mu_);
parallelism = num_parallel_calls_->value;
}
// NOTE: We do not synchronize the following access to
// num_parallel_calls_ to minimize the tracing overhead.
int64 parallelism = num_parallel_calls_->value;
return strings::StrCat(prefix(), "#parallelism=", parallelism, "#");
}

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

@ -213,11 +213,9 @@ class ParallelInterleaveDatasetOp : public UnaryDatasetOpKernel {
}
string BuildTraceMeName() override {
int64 parallelism;
{
tf_shared_lock l(*mu_);
parallelism = num_parallel_calls_->value;
}
// NOTE: We do not synchronize the following access to
// num_parallel_calls_ to minimize the tracing overhead.
int64 parallelism = num_parallel_calls_->value;
return strings::StrCat(prefix(), "#parallelism=", parallelism, "#");
}

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

@ -74,11 +74,9 @@ class ParallelMapIterator : public DatasetBaseIterator {
}
string BuildTraceMeName() override {
int64 parallelism;
{
tf_shared_lock l(*mu_);
parallelism = num_parallel_calls_->value;
}
// NOTE: We do not synchronize the following access to num_parallel_calls_
// to minimize the tracing overhead.
int64 parallelism = num_parallel_calls_->value;
return strings::StrCat(prefix(), "#parallelism=", parallelism, "#");
}