From 054f3638ed995ba7b2345dd757fb855ea65eaf89 Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Thu, 11 Mar 2021 21:46:49 -0800
Subject: [PATCH] Internal change

PiperOrigin-RevId: 362447681
---
 keras/benchmarks/BUILD                        | 14 +++++++++++
 keras/benchmarks/layer_benchmarks/BUILD       |  1 +
 keras/benchmarks/saved_model_benchmarks/BUILD |  8 +++++++
 keras/integration_test/BUILD                  |  9 ++++++++
 keras/tests/BUILD                             | 23 ++++++++++++++++++-
 5 files changed, 54 insertions(+), 1 deletion(-)

diff --git a/keras/benchmarks/BUILD b/keras/benchmarks/BUILD
index 6a66ac89754b..20ba29b2b7f2 100644
--- a/keras/benchmarks/BUILD
+++ b/keras/benchmarks/BUILD
@@ -64,6 +64,7 @@ py_test(
         ":profiler_lib",
         "//:expect_numpy_installed",
         "//:expect_tensorflow_installed",
+        "//keras/api:keras_api",
     ],
 )
 
@@ -78,6 +79,7 @@ cuda_py_test(
     deps = [
         ":profiler_lib",
         "//:expect_tensorflow_installed",
+        "//keras/api:keras_api",
         "//keras/utils:tf_inspect",
     ],
 )
@@ -89,6 +91,7 @@ cuda_py_test(
     deps = [
         ":profiler_lib",
         "//:expect_tensorflow_installed",
+        "//keras/api:keras_api",
     ],
 )
 
@@ -111,6 +114,7 @@ py_test(
     deps = [
         ":benchmark_util",
         "//:expect_tensorflow_installed",
+        "//keras/api:keras_api",
     ],
 )
 
@@ -123,6 +127,7 @@ cuda_py_test(
         ":benchmark_util",
         ":profiler_lib",
         "//:expect_tensorflow_installed",
+        "//keras/api:keras_api",
     ],
 )
 
@@ -135,6 +140,7 @@ cuda_py_test(
         ":benchmark_util",
         ":profiler_lib",
         "//:expect_tensorflow_installed",
+        "//keras/api:keras_api",
     ],
 )
 
@@ -147,6 +153,7 @@ cuda_py_test(
         ":benchmark_util",
         ":profiler_lib",
         "//:expect_tensorflow_installed",
+        "//keras/api:keras_api",
     ],
 )
 
@@ -160,6 +167,7 @@ cuda_py_test(
         ":profiler_lib",
         "//:expect_numpy_installed",
         "//:expect_tensorflow_installed",
+        "//keras/api:keras_api",
     ],
 )
 
@@ -172,6 +180,7 @@ cuda_py_test(
         ":benchmark_util",
         ":profiler_lib",
         "//:expect_tensorflow_installed",
+        "//keras/api:keras_api",
     ],
 )
 
@@ -184,6 +193,7 @@ cuda_py_test(
         ":benchmark_util",
         ":profiler_lib",
         "//:expect_tensorflow_installed",
+        "//keras/api:keras_api",
     ],
 )
 
@@ -197,6 +207,7 @@ cuda_py_test(
         ":profiler_lib",
         "//:expect_numpy_installed",
         "//:expect_tensorflow_installed",
+        "//keras/api:keras_api",
     ],
 )
 
@@ -209,6 +220,7 @@ cuda_py_test(
         ":benchmark_util",
         ":profiler_lib",
         "//:expect_tensorflow_installed",
+        "//keras/api:keras_api",
     ],
 )
 
@@ -222,6 +234,7 @@ cuda_py_test(
         ":distribution_util",
         ":profiler_lib",
         "//:expect_tensorflow_installed",
+        "//keras/api:keras_api",
     ],
 )
 
@@ -245,6 +258,7 @@ py_test(
         ":benchmark_util",
         ":profiler_lib",
         "//:expect_tensorflow_installed",
+        "//keras/api:keras_api",
         "//keras/optimizer_v2",
     ],
 )
diff --git a/keras/benchmarks/layer_benchmarks/BUILD b/keras/benchmarks/layer_benchmarks/BUILD
index 0a80a1a6136a..78e28841919d 100644
--- a/keras/benchmarks/layer_benchmarks/BUILD
+++ b/keras/benchmarks/layer_benchmarks/BUILD
@@ -64,6 +64,7 @@ tf_py_test(
     deps = [
         ":layer_benchmarks_test_base",
         "//:expect_tensorflow_installed",
+        "//keras/api:keras_api",
         "//keras/benchmarks:benchmark_util",
     ],
 )
diff --git a/keras/benchmarks/saved_model_benchmarks/BUILD b/keras/benchmarks/saved_model_benchmarks/BUILD
index 90929a0c76b7..0a4f380fb165 100644
--- a/keras/benchmarks/saved_model_benchmarks/BUILD
+++ b/keras/benchmarks/saved_model_benchmarks/BUILD
@@ -47,6 +47,7 @@ cuda_py_test(
     deps = [
         ":saved_model_benchmark_util",
         "//:expect_tensorflow_installed",
+        "//keras/api:keras_api",
         "//keras/benchmarks:profiler_lib",
     ],
 )
@@ -61,6 +62,7 @@ cuda_py_test(
     deps = [
         ":saved_model_benchmark_util",
         "//:expect_tensorflow_installed",
+        "//keras/api:keras_api",
         "//keras/benchmarks:profiler_lib",
     ],
 )
@@ -75,6 +77,7 @@ cuda_py_test(
     deps = [
         ":saved_model_benchmark_util",
         "//:expect_tensorflow_installed",
+        "//keras/api:keras_api",
         "//keras/benchmarks:profiler_lib",
     ],
 )
@@ -89,6 +92,7 @@ cuda_py_test(
     deps = [
         ":saved_model_benchmark_util",
         "//:expect_tensorflow_installed",
+        "//keras/api:keras_api",
         "//keras/benchmarks:profiler_lib",
     ],
 )
@@ -103,6 +107,7 @@ cuda_py_test(
     deps = [
         ":saved_model_benchmark_util",
         "//:expect_tensorflow_installed",
+        "//keras/api:keras_api",
         "//keras/benchmarks:profiler_lib",
     ],
 )
@@ -117,6 +122,7 @@ cuda_py_test(
     deps = [
         ":saved_model_benchmark_util",
         "//:expect_tensorflow_installed",
+        "//keras/api:keras_api",
         "//keras/benchmarks:profiler_lib",
     ],
 )
@@ -131,6 +137,7 @@ cuda_py_test(
     deps = [
         ":saved_model_benchmark_util",
         "//:expect_tensorflow_installed",
+        "//keras/api:keras_api",
         "//keras/benchmarks:profiler_lib",
     ],
 )
@@ -145,6 +152,7 @@ cuda_py_test(
     deps = [
         ":saved_model_benchmark_util",
         "//:expect_tensorflow_installed",
+        "//keras/api:keras_api",
         "//keras/benchmarks:profiler_lib",
     ],
 )
diff --git a/keras/integration_test/BUILD b/keras/integration_test/BUILD
index 00b51df2407f..0c0298ba62f8 100644
--- a/keras/integration_test/BUILD
+++ b/keras/integration_test/BUILD
@@ -19,6 +19,7 @@ tf_py_test(
     deps = [
         "//:expect_absl_installed",
         "//:expect_tensorflow_installed",
+        "//keras/api:keras_api",
     ],
 )
 
@@ -28,6 +29,7 @@ tf_py_test(
     python_version = "PY3",
     deps = [
         "//:expect_tensorflow_installed",
+        "//keras/api:keras_api",
     ],
 )
 
@@ -37,6 +39,7 @@ tf_py_test(
     python_version = "PY3",
     deps = [
         "//:expect_tensorflow_installed",
+        "//keras/api:keras_api",
     ],
 )
 
@@ -47,6 +50,7 @@ cuda_py_test(
     deps = [
         "//:expect_absl_installed",
         "//:expect_tensorflow_installed",
+        "//keras/api:keras_api",
     ],
 )
 
@@ -56,6 +60,7 @@ tf_py_test(
     python_version = "PY3",
     deps = [
         "//:expect_tensorflow_installed",
+        "//keras/api:keras_api",
     ],
 )
 
@@ -64,6 +69,7 @@ tf_py_test(
     srcs = ["module_test.py"],
     deps = [
         "//:expect_tensorflow_installed",
+        "//keras/api:keras_api",
     ],
 )
 
@@ -73,6 +79,7 @@ tf_py_test(
     python_version = "PY3",
     deps = [
         "//:expect_tensorflow_installed",
+        "//keras/api:keras_api",
     ],
 )
 
@@ -82,6 +89,7 @@ cuda_py_test(
     python_version = "PY3",
     deps = [
         "//:expect_tensorflow_installed",
+        "//keras/api:keras_api",
     ],
 )
 
@@ -131,5 +139,6 @@ tf_py_test(
         "//:expect_absl_installed",
         "//:expect_portpicker_installed",
         "//:expect_tensorflow_installed",
+        "//keras/api:keras_api",
     ],
 )
diff --git a/keras/tests/BUILD b/keras/tests/BUILD
index 0a4a5a503be4..aea97aeb4b3c 100644
--- a/keras/tests/BUILD
+++ b/keras/tests/BUILD
@@ -33,6 +33,7 @@ tf_py_test(
         "//:expect_absl_installed",
         "//:expect_tensorflow_installed",
         "//keras",
+        "//keras/api:keras_api",
     ],
 )
 
@@ -46,6 +47,7 @@ tf_py_test(
         "//:expect_numpy_installed",
         "//:expect_tensorflow_installed",
         "//keras",
+        "//keras/api:keras_api",
     ],
 )
 
@@ -62,6 +64,7 @@ tpu_py_test(
         "//:expect_tensorboard_installed",
         "//:expect_tensorflow_installed",
         "//keras",
+        "//keras/api:keras_api",
         "//keras/distribute:distribute_strategy_test_lib",
     ],
 )
@@ -75,6 +78,7 @@ tf_py_test(
         "//:expect_tensorflow_installed",
         "//keras",
         "//keras:testing_utils",
+        "//keras/api:keras_api",
     ],
 )
 
@@ -89,6 +93,7 @@ tf_py_test(
         "//:expect_numpy_installed",
         "//:expect_tensorflow_installed",
         "//keras",
+        "//keras/api:keras_api",
     ],
 )
 
@@ -99,6 +104,7 @@ tf_py_test(
     deps = [
         "//:expect_tensorflow_installed",
         "//keras",
+        "//keras/api:keras_api",
     ],
 )
 
@@ -116,6 +122,7 @@ tf_py_test(
         "//:expect_numpy_installed",
         "//:expect_tensorflow_installed",
         "//keras",
+        "//keras/api:keras_api",
         "//keras/layers/legacy_rnn:rnn_cell_impl",
         "//keras/legacy_tf_layers:layers_base",
     ],
@@ -142,6 +149,7 @@ tf_py_test(
         "//:expect_absl_installed",
         "//:expect_numpy_installed",
         "//keras",
+        "//keras/api:keras_api",
     ],
 )
 
@@ -170,6 +178,7 @@ tf_py_test(
         "//:expect_tensorflow_installed",
         "//keras",
         "//keras:combinations",
+        "//keras/api:keras_api",
     ],
 )
 
@@ -188,6 +197,7 @@ tf_py_test(
         "//:expect_numpy_installed",
         "//:expect_tensorflow_installed",
         "//keras",
+        "//keras/api:keras_api",
     ],
 )
 
@@ -207,6 +217,7 @@ cuda_py_test(
         "//:expect_six_installed",
         "//:expect_tensorflow_installed",
         "//keras",
+        "//keras/api:keras_api",
     ],
 )
 
@@ -224,7 +235,10 @@ tf_py_test(
         "nomsan",  # TODO(b/149948895): Re-enable.
         "notsan",  # TODO(b/149948895): Re-enable.
     ],
-    deps = ["//:expect_tensorflow_installed"],
+    deps = [
+        "//:expect_tensorflow_installed",
+        "//keras/api:keras_api",
+    ],
 )
 
 tf_py_test(
@@ -239,6 +253,7 @@ tf_py_test(
         "//:expect_tensorflow_installed",
         "//keras",
         "//keras:metrics",
+        "//keras/api:keras_api",
         "//keras/layers:core",
         "//keras/optimizer_v2",
     ],
@@ -251,6 +266,7 @@ cuda_py_test(
     python_version = "PY3",
     deps = [
         "//:expect_tensorflow_installed",
+        "//keras/api:keras_api",
         "//keras/engine",
         "//keras/layers:core",
     ],
@@ -265,6 +281,7 @@ tf_py_test(
         "//:expect_tensorflow_installed",
         "//keras",
         "//keras:combinations",
+        "//keras/api:keras_api",
         "//keras/engine",
         "//keras/layers:core",
     ],
@@ -279,6 +296,7 @@ tf_py_test(
         "//:expect_numpy_installed",
         "//:expect_tensorflow_installed",
         "//keras",
+        "//keras/api:keras_api",
     ],
 )
 
@@ -295,6 +313,7 @@ tf_py_test(
         "//:expect_tensorflow_installed",
         "//keras",
         "//keras:combinations",
+        "//keras/api:keras_api",
         "//keras/engine",
         "//keras/layers:core",
         "//keras/layers:normalization",
@@ -311,6 +330,7 @@ tf_py_test(
         "//:expect_tensorflow_installed",
         "//keras",
         "//keras:combinations",
+        "//keras/api:keras_api",
         "//keras/engine",
         "//keras/layers:core",
         "//keras/optimizer_v2",
@@ -327,6 +347,7 @@ tf_py_test(
         "//:expect_tensorflow_installed",
         "//keras",
         "//keras:combinations",
+        "//keras/api:keras_api",
         "//keras/engine",
         "//keras/layers:core",
     ],