Use ubuntu2004 for TensorFlow (#1368)

Fixing https://github.com/tensorflow/tensorflow/issues/56118
1 file changed