message: Run the grappler optimizer tests both on GPU and CPU
diff: licenses(["notice"]) # Apache 2.0
  load("//tensorflow:tensorflow.bzl", "tf_cc_test")
  -load("//tensorflow:tensorflow.bzl", "tf_cc_test_gpu")
  +load("//tensorflow:tensorflow.bzl", "tf_cuda_cc_test")
  load("//tensorflow:tensorflow.bzl", "tf_kernel_library")
  load("@local_config_cuda//cuda:build_defs.bzl", "if_cuda")
  @@ -44,7 +44,7 @@ cc_library(
  ],
  )
  -tf_cc_test(
  +tf_cuda_cc_test(
  name = "static_schedule_test",
  srcs = ["static_schedule_test.cc"],
  deps = [
  @@ -79,7 +79,7 @@ cc_library(
  ],
  )
  -tf_cc_test(
  +tf_cuda_cc_test(
  name = "auto_parallel_test",
  srcs = ["auto_parallel_test.cc"],
  deps = [
  @@ -157,7 +157,7 @@ cc_library(
  ],
  )
  -tf_cc_test(
  +tf_cuda_cc_test(
  name = "function_optimizer_test",
  srcs = ["function_optimizer_test.cc"],
  deps = [
  @@ -223,7 +223,7 @@ cc_library(
  ],
  )
  -tf_cc_test(
  +tf_cuda_cc_test(
  name = "graph_optimizer_stage_test",
  size = "small",
  srcs = ["graph_optimizer_stage_test.cc"],
  @@ -274,7 +274,7 @@ cc_library(
  ],
  )
  -tf_cc_test(
  +tf_cuda_cc_test(
  name = "arithmetic_optimizer_test",
  size = "small",
  srcs = ["arithmetic_optimizer_test.cc"],
  @@ -315,7 +315,7 @@ cc_library(
  ],
  )
  -tf_cc_test(
  +tf_cuda_cc_test(
  name = "dependency_optimizer_test",
  size = "small",
  srcs = ["dependency_optimizer_test.cc"],
  @@ -351,7 +351,7 @@ cc_library(
  ],
  )
  -tf_cc_test(
  +tf_cuda_cc_test(
  name = "model_pruner_test",
  srcs = ["model_pruner_test.cc"],
  deps = [
  @@ -422,10 +422,9 @@ cc_library(
  ]),
  )
  -tf_cc_test_gpu(
  +tf_cuda_cc_test(
  name = "memory_optimizer_test",
  srcs = ["memory_optimizer_test.cc"],
  - tags = ["no_cuda_on_cpu_tap"],
  deps = [
  ":memory_optimizer",
  "//tensorflow/cc:cc_ops",
  @@ -464,7 +463,7 @@ cc_library(
  ],
  )
  -tf_cc_test(
  +tf_cuda_cc_test(
  name = "layout_optimizer_test",
  srcs = ["layout_optimizer_test.cc"],
  deps = [
  @@ -513,7 +512,7 @@ cc_library(
  ],
  )
  -tf_cc_test(
  +tf_cuda_cc_test(
  name = "meta_optimizer_test",
  srcs = ["meta_optimizer_test.cc"],
  deps = [
  @@ -542,7 +541,7 @@ cc_library(
  ],
  )
  -tf_cc_test(
  +tf_cuda_cc_test(
  name = "custom_graph_optimizer_registry_test",
  size = "small",
  srcs = ["custom_graph_optimizer_registry_test.cc"],
  @@ -577,7 +576,7 @@ cc_library(
  ],
  )
  -tf_cc_test(
  +tf_cuda_cc_test(
  name = "loop_optimizer_test",
  srcs = ["loop_optimizer_test.cc"],
  deps = [
  