Model Training Performance difference between using TF.data vs NumPy for data loading

Question

I am trying to reimplement the data loading and feeding to the model for training functionality as shown below using tf.data as compared to this original implementation wher