message: [XLA] Update test timeouts so the tests pass with optimized + -UNDEBUG
diff: @@ -251,6 +251,7 @@ tf_xla_py_test(
  tf_xla_py_test(
  name = "matrix_triangular_solve_op_test",
  size = "small",
  + timeout = "moderate",
  srcs = ["matrix_triangular_solve_op_test.py"],
  tags = ["optonly"],
  deps = [
  @@ -572,6 +573,7 @@ tf_xla_py_test(
  tf_xla_py_test(
  name = "matrix_band_part_test",
  size = "medium",
  + timeout = "long",
  srcs = ["matrix_band_part_test.py"],
  tags = ["optonly"],
  deps = [
  