Como saber a duração de um conjunto de tensorflow do conjunto de dados

ds = tf.data.TFRecordDataset(dataset_filenames)
ds_size = sum(1 for _ in ds)
javimu