Fix env check order by splitting TPU envs into separate files.

Fix env check order by splitting TPU envs into separate files. #2482