Add TensorFlow version to `TPU_ML_PLATFORM_VERSION` environment variables.
Add TensorFlow version to `TPU_ML_PLATFORM_VERSION` environment variables.

This will allow us to track the version of TensorFlow that is being used to run on Cloud TPU.
