TensorFlow 2 version | View source on GitHub |
Records the latency of producing each element of the input dataset.
tf.data.experimental.latency_stats(
tag
)
To consume the statistics, associate a StatsAggregator
with the output
dataset.
Args | |
---|---|
tag
|
String. All statistics recorded by the returned transformation will
be associated with the given tag .
|
Returns | |
---|---|
A Dataset transformation function, which can be passed to
tf.data.Dataset.apply .
|