message: Double test timeouts for "small" and "medium" tests to account for Cloud overhead.
diff: @@ -54,6 +54,7 @@ bazel build "${bazel_args[@]}" -- "${test_patterns[@]}"
  ctpu_up -s v2-8 -p tensorflow-testing-tpu
  test_args=(
  + --test_timeout=120,600,-1,-1 \
  --test_arg=--tpu="${TPU_NAME}" \
  --test_arg=--zone="${TPU_ZONE}" \
  --test_arg=--test_dir_base=gs://kokoro-tpu-testing/tempdir/ \
  