tf.data.experimental.StatsAggregator