Tensorflow splitting dataset into training and testing causes bottleneck/slow

Question

I have a dataset, and when I do preprocessing on it with ds = ds.map(process_path, num_parallel_calls=AUTOTUNE).prefetch(AUTOTUNE) that line executes very fast.