Use ubuntu2004 for TensorFlow (#1368) Fixing https://github.com/tensorflow/tensorflow/issues/56118
See https://github.com/bazelbuild/continuous-integration/blob/master/buildkite/README.md