message: Increase shard count for test that times out
diff: @@ -1119,6 +1119,7 @@ cuda_py_test(
  size = "small",
  srcs = ["device_placement_test.py"],
  python_version = "PY3",
  + shard_count = 5,
  deps = [
  ":context",
  ":def_function",
  