diff --git a/tensorflow/python/data/experimental/__init__.py b/tensorflow/python/data/experimental/__init__.py index 275bdf7ef44b1012bf443e23644ee71df9d31709..f7103a621ee09bad322a5add73d81308b6b2be8c 100644 --- a/tensorflow/python/data/experimental/__init__.py +++ b/tensorflow/python/data/experimental/__init__.py @@ -42,6 +42,7 @@ See [Importing Data](https://tensorflow.org/guide/datasets) for an overview. @@ThreadingOptions @@bucket_by_sequence_length +@@bytes_produced_stats @@cardinality @@choose_from_datasets @@copy_to_device @@ -116,6 +117,7 @@ from tensorflow.python.data.experimental.ops.resampling import rejection_resampl from tensorflow.python.data.experimental.ops.scan_ops import scan from tensorflow.python.data.experimental.ops.shuffle_ops import shuffle_and_repeat from tensorflow.python.data.experimental.ops.stats_aggregator import StatsAggregator +from tensorflow.python.data.experimental.ops.stats_ops import bytes_produced_stats from tensorflow.python.data.experimental.ops.stats_ops import latency_stats from tensorflow.python.data.experimental.ops.stats_options import StatsOptions from tensorflow.python.data.experimental.ops.take_while_ops import take_while diff --git a/tensorflow/python/data/experimental/ops/stats_ops.py b/tensorflow/python/data/experimental/ops/stats_ops.py index 13dcb92fa0643c0f89110307f2c13cb6e8425a56..dff4286dabffcac8f3274d8e45e07f9a7a284fe3 100644 --- a/tensorflow/python/data/experimental/ops/stats_ops.py +++ b/tensorflow/python/data/experimental/ops/stats_ops.py @@ -48,8 +48,7 @@ def set_stats_aggregator(stats_aggregator, prefix="", counter_prefix=""): return _apply_fn -# TODO(b/38416882): Properly export in the `tf.data.experimental` API when -# stable or make private / remove. +@tf_export("data.experimental.bytes_produced_stats") def bytes_produced_stats(tag): """Records the number of bytes produced by each element of the input dataset. diff --git a/tensorflow/tools/api/golden/v1/tensorflow.data.experimental.pbtxt b/tensorflow/tools/api/golden/v1/tensorflow.data.experimental.pbtxt index 0e3999c0f4618f2b09a23ea0a32b5ae89990258d..853054aab06f4e67e1b489269a74cf1418c0bf4c 100644 --- a/tensorflow/tools/api/golden/v1/tensorflow.data.experimental.pbtxt +++ b/tensorflow/tools/api/golden/v1/tensorflow.data.experimental.pbtxt @@ -88,6 +88,10 @@ tf_module { name: "bucket_by_sequence_length" argspec: "args=[\'element_length_func\', \'bucket_boundaries\', \'bucket_batch_sizes\', \'padded_shapes\', \'padding_values\', \'pad_to_bucket_boundary\', \'no_padding\', \'drop_remainder\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'False\', \'False\', \'False\'], " } + member_method { + name: "bytes_produced_stats" + argspec: "args=[\'tag\'], varargs=None, keywords=None, defaults=None" + } member_method { name: "cardinality" argspec: "args=[\'dataset\'], varargs=None, keywords=None, defaults=None" diff --git a/tensorflow/tools/api/golden/v2/tensorflow.data.experimental.pbtxt b/tensorflow/tools/api/golden/v2/tensorflow.data.experimental.pbtxt index 48d90028646da07d95478abc9a3ddb5e918599e6..08b7657a5f1e87ca242abfd5157972a31ba5b1bc 100644 --- a/tensorflow/tools/api/golden/v2/tensorflow.data.experimental.pbtxt +++ b/tensorflow/tools/api/golden/v2/tensorflow.data.experimental.pbtxt @@ -88,6 +88,10 @@ tf_module { name: "bucket_by_sequence_length" argspec: "args=[\'element_length_func\', \'bucket_boundaries\', \'bucket_batch_sizes\', \'padded_shapes\', \'padding_values\', \'pad_to_bucket_boundary\', \'no_padding\', \'drop_remainder\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'False\', \'False\', \'False\'], " } + member_method { + name: "bytes_produced_stats" + argspec: "args=[\'tag\'], varargs=None, keywords=None, defaults=None" + } member_method { name: "cardinality" argspec: "args=[\'dataset\'], varargs=None, keywords=None, defaults=None"