Using tf.function gives you performance as well as reusability benefits by using same traces and gives you flexibility to use concrete functions.
Attaching doc reference 7 for how better performance can be achieved with tf.function.