message: Fix the BUILD file
diff: @@ -44,6 +44,27 @@ cuda_py_test(
  ],
  )
  +cuda_py_test(
  + name = "applications_saved_model_test",
  + size = "medium",
  + srcs = ["applications_saved_model_test.py"],
  + shard_count = 8,
  + deps = [
  + "//tensorflow/python:client_testlib",
  + "//tensorflow/python/keras/applications",
  + "@absl_py//absl/testing:parameterized",
  + ],
  +)
  +
  +cuda_py_test(
  + name = "model_components_benchmarks_test",
  + srcs = ["model_components_benchmarks_test.py"],
  + python_version = "PY3",
  + deps = [
  + "//tensorflow:tensorflow_py",
  + ],
  +)
  +
  py_test(
  name = "keras_examples_benchmark_test",
  size = "medium",
  @@ -69,30 +90,3 @@ py_library(
  "//tensorflow/python/eager:remote",
  ],
  )
  \ No newline at end of file
  -
  -cuda_py_test(
  - name = "applications_saved_model_test",
  - size = "medium",
  - srcs = ["applications_saved_model_test.py"],
  - shard_count = 8,
  - deps = [
  - "//tensorflow/python:client_testlib",
  - "//tensorflow/python/keras/applications",
  - "@absl_py//absl/testing:parameterized",
  - ],
  -)
  -
  -cuda_py_test(
  - name = "model_components_benchmarks_test",
  - srcs = ["model_components_benchmarks_test.py"],
  - python_version = "PY3",
  - deps = [
  - "//tensorflow/python:random_ops",
  - "//tensorflow/python:training_lib",
  - "//tensorflow/python/data/ops:dataset_ops",
  - "//tensorflow/python/eager:backprop",
  - "//tensorflow/python/eager:context",
  - "//tensorflow/python/eager:profiler",
  - "//tensorflow/python/eager:test",
  - ],
  -)
  