[tf.data] Add an attr to limit the max parallelism of map defun