#tfdata Set code-based default name for map and parallel map transformations.
#tfdata Set code-based default name for map and parallel map transformations.

This is helpful for correlating tf.data maps in profiling traces with the code that generated them.

FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/openxla/xla/pull/9778 from shraiysh:flash-attn-2 77a644020c82939ea89c717c3490b5469d35e175
