I am using tf.keras.preprocessing.image_dataset_from_directory.
tf.keras.preprocessing.image_dataset_from_directory
According to the documentation, the related image_size parameter is the
image_size