message: This test can run for a considerable amount of time. Avoid timeouts.
diff: @@ -3015,6 +3015,7 @@ cuda_py_test(
  cuda_py_test(
  name = "rnn_test",
  size = "medium",
  + timeout = "long",
  srcs = ["rnn_test.py"],
  additional_deps = [
  "//third_party/py/numpy",
  