That’s the right link, thanks @Bhack.
The main differences are:


tf.io.TFRecordWriter is a python interface where you call writer.write(example).

tf.data.experimental.TFRecordWriter is experimental, and writes out a whole tf.data.Dataset. If you already have your data in a tf.data.Dataset, that may be simpler.