message: Up the shard_count to help reduce_test on the edge of timing out
diff: @@ -699,6 +699,7 @@ tf_py_test(
  name = "reduction_test",
  srcs = ["reduction_test.py"],
  python_version = "PY3",
  + shard_count = 4,
  tags = ["notsan"], # TODO(b/170783154)
  deps = [
  ":reduction",
  