tf.io.TFRecordWriter