Tensorflow element wise gradient slower when using tf.function

Question

I need to compute an elementwise gradient as input for a neural network. I decided to use a tf.data.Dataset to store the input data. Preprocessing the data and comp