Tensorflow Dataset Split Images into Tiles

Question

I am loading my dataset as follows,

ds =  tf.data.Dataset.list_files("/images/*.png")

train_size = int(0.8 * len(ds))
train_ds = ds.take(train_size