message: Use distribute_py_test instead of tpu_py_test for preprocessing distribute tests
  These hit a wider range of strategies in the tests
diff: @@ -5,7 +5,6 @@ load("//tensorflow:tensorflow.bzl", "tf_py_test")
  # buildifier: disable=same-origin-load
  load("//tensorflow:tensorflow.bzl", "cuda_py_test")
  -load("//tensorflow/python/tpu:tpu.bzl", "tpu_py_test")
  load("//tensorflow/core/platform/default:distribute.bzl", "distribute_py_test")
  package(
  @@ -517,13 +516,15 @@ cuda_py_test(
  ],
  )
  -tpu_py_test(
  +distribute_py_test(
  name = "hashing_distribution_test",
  srcs = ["hashing_distribution_test.py"],
  disable_mlir_bridge = False,
  main = "hashing_distribution_test.py",
  python_version = "PY3",
  - tags = ["multi_and_single_gpu"],
  + tags = [
  + "multi_and_single_gpu",
  + ],
  deps = [
  ":hashing",
  "//tensorflow/python/distribute:combinations",
  @@ -548,13 +549,16 @@ tf_py_test(
  ],
  )
  -tpu_py_test(
  +distribute_py_test(
  name = "index_lookup_distribution_test",
  srcs = ["index_lookup_distribution_test.py"],
  disable_mlir_bridge = False,
  main = "index_lookup_distribution_test.py",
  python_version = "PY3",
  - tags = ["no_oss"],
  + tags = [
  + "multi_and_single_gpu",
  + ],
  + tpu_tags = ["no_oss"],
  deps = [
  ":index_lookup",
  "//tensorflow/python/distribute:combinations",
  