Add ability to take TPU address and output dir from environment variables.
Add ability to take TPU address and output dir from environment variables.

PiperOrigin-RevId: 300858086

tombstone/models,tombstone/models,tombstone/models,tombstone/models,tombstone/models,tombstone/models