r2.16 cherry-pick: 20a8fc4339d "Inherit env variables for tpu grpc worker console script"
Refer to the original commit: https://github.com/tensorflow/tensorflow/commit/20a8fc4339db1ef98c537ee368c705bbd9ddaa45