diff options
Diffstat (limited to 'tensorflow/core/framework/stats_aggregator.h')
| -rw-r--r-- | tensorflow/core/framework/stats_aggregator.h | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/core/framework/stats_aggregator.h b/tensorflow/core/framework/stats_aggregator.h index 8002d9291c..4a18efc940 100644 --- a/tensorflow/core/framework/stats_aggregator.h +++ b/tensorflow/core/framework/stats_aggregator.h @@ -57,6 +57,10 @@ class StatsAggregator { // interface. It is possible that not all implementations will support // encoding their state as a protocol buffer. virtual void EncodeToProto(Summary* out_summary) = 0; + + // Increment the `label` cell of metrics mapped with `name` by given `value`. + virtual void IncrementCounter(const string& name, const string& label, + int64 val) = 0; }; // A `StatsAggregatorResource` wraps a shareable `StatsAggregator` as a resource |