Internal change

PiperOrigin-RevId: 362447681

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",
     ],
 )

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",
     ],
 )

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",
     ],
 )

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",
     ],
 )