diff options
author | Shivani Agrawal <shivaniagrawal@google.com> | 2018-10-03 12:44:47 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-03 12:51:54 -0700 |
commit | 808b1dcb318b1feb5a8c9fed5558f95cd05728e4 (patch) | |
tree | a2286241b6a0c8cba24b1da629fa6e7db475d7dc /tensorflow/core/ops | |
parent | 19833284cc8fa555115aacde350ad66652b250dc (diff) |
[data-stats] Sets user given `tag` and `counter_prefix` with `set_stats_aggregator`. `tag` would get prep-end with all the statistics recorded as summary and `counter_prefix` would set the prefix for the statistics recorded as counter.
Note: `counter` defaults to `\tensorflow`, and `tag` and `prefix` gets associated with the dataset (not the stats_aggregator).
PiperOrigin-RevId: 215609159
Diffstat (limited to 'tensorflow/core/ops')
-rw-r--r-- | tensorflow/core/ops/compat/ops_history.v1.pbtxt | 8 | ||||
-rw-r--r-- | tensorflow/core/ops/dataset_ops.cc | 2 |
2 files changed, 10 insertions, 0 deletions
diff --git a/tensorflow/core/ops/compat/ops_history.v1.pbtxt b/tensorflow/core/ops/compat/ops_history.v1.pbtxt index 4845767405..33f18ae13f 100644 --- a/tensorflow/core/ops/compat/ops_history.v1.pbtxt +++ b/tensorflow/core/ops/compat/ops_history.v1.pbtxt @@ -59785,6 +59785,14 @@ op { name: "stats_aggregator" type: DT_RESOURCE } + input_arg { + name: "tag" + type: DT_STRING + } + input_arg { + name: "counter_prefix" + type: DT_STRING + } output_arg { name: "handle" type: DT_VARIANT diff --git a/tensorflow/core/ops/dataset_ops.cc b/tensorflow/core/ops/dataset_ops.cc index 71f4cc3c4c..889a6a4640 100644 --- a/tensorflow/core/ops/dataset_ops.cc +++ b/tensorflow/core/ops/dataset_ops.cc @@ -185,6 +185,8 @@ REGISTER_OP("ParseExampleDataset") REGISTER_OP("SetStatsAggregatorDataset") .Input("input_dataset: variant") .Input("stats_aggregator: resource") + .Input("tag: string") + .Input("counter_prefix: string") .Output("handle: variant") .Attr("output_types: list(type) >= 1") .Attr("output_shapes: list(shape) >= 1") |