Fix some keras test that lack of explicit build deps.

PiperOrigin-RevId: 336160614
Change-Id: I0f89f1b81f1ba3f2e362489042a361040e381307

tensorflow/python/keras/tests/BUILD

@@ -103,6 +103,8 @@ tf_py_test(
         "//tensorflow/python:tensor_spec",
         "//tensorflow/python:util",
         "//tensorflow/python/eager:def_function",
+        "//tensorflow/python/keras",
+        "//tensorflow/python/keras:testing_utils",
         "//tensorflow/python/saved_model:load",
         "//tensorflow/python/saved_model:save",
         "//tensorflow/python/training/tracking",
@@ -137,6 +139,7 @@ tf_py_test(
         "//tensorflow/python:saver",
         "//tensorflow/python:session",
         "//tensorflow/python:tf_optimizer",
+        "//tensorflow/python/keras",
     ],
 )
 
@@ -154,6 +157,7 @@ tf_py_test(
         "//tensorflow/python:client_testlib",
         "//tensorflow/python:nn_ops",
         "//tensorflow/python/keras",
+        "//tensorflow/python/keras/layers/legacy_rnn:rnn_cell_impl",
         "//tensorflow/python/keras/legacy_tf_layers:layers_base",
         "//third_party/py/numpy",
         "@absl_py//absl/testing:parameterized",
@@ -292,6 +296,7 @@ cuda_py_test(
         "//tensorflow/python:util",
         "//tensorflow/python/eager:context",
         "//tensorflow/python/eager:test",
+        "//tensorflow/python/keras",
         "//tensorflow/python/keras:combinations",
     ],
 )
@@ -308,6 +313,7 @@ cuda_py_test(
         "//tensorflow/python:lib",
         "//tensorflow/python:platform",
         "//tensorflow/python:summary_ops_v2",
+        "//tensorflow/python/keras:testing_utils",
         "//tensorflow/python/keras/engine",
         "//tensorflow/python/keras/layers:core",
     ],
@@ -327,6 +333,8 @@ tf_py_test(
         "//tensorflow/python:tensor_spec",
         "//tensorflow/python/eager:backprop",
         "//tensorflow/python/eager:function",
+        "//tensorflow/python/keras",
+        "//tensorflow/python/keras:metrics",
         "//tensorflow/python/keras/layers:core",
         "//tensorflow/python/keras/optimizer_v2",
         "//tensorflow/python/saved_model:save",
@@ -368,6 +376,7 @@ tf_py_test(
         "//tensorflow/python:constant_op",
         "//tensorflow/python:framework_test_lib",
         "//tensorflow/python:util",
+        "//tensorflow/python/keras",
         "//tensorflow/python/keras:combinations",
         "//tensorflow/python/keras/engine",
         "//tensorflow/python/keras/layers:core",
@@ -405,6 +414,7 @@ tf_py_test(
         "//tensorflow/python:variables",
         "//tensorflow/python/eager:context",
         "//tensorflow/python/eager:test",
+        "//tensorflow/python/keras",
         "//tensorflow/python/keras:combinations",
         "//tensorflow/python/keras/engine",
         "//tensorflow/python/keras/layers:core",

tensorflow/python/keras/tests/integration_test.py

@@ -28,10 +28,10 @@ from tensorflow.python.eager import context
 from tensorflow.python.framework import dtypes
 from tensorflow.python.keras import keras_parameterized
 from tensorflow.python.keras import testing_utils
+from tensorflow.python.keras.layers.legacy_rnn import rnn_cell_impl as rnn_cell
 from tensorflow.python.keras.legacy_tf_layers import base as base_layer
 from tensorflow.python.keras.utils import np_utils
 from tensorflow.python.ops import nn_ops as nn
-from tensorflow.python.ops import rnn_cell
 from tensorflow.python.platform import test
 
 

tensorflow/python/keras/utils/BUILD

@@ -314,6 +314,7 @@ tf_py_test(
         "//tensorflow/python:math_ops",
         "//tensorflow/python:sparse_ops",
         "//tensorflow/python:sparse_tensor",
+        "//tensorflow/python/keras",
         "//tensorflow/python/keras:engine",
         "//tensorflow/python/keras/layers",
         "//tensorflow/python/ops/ragged:ragged_tensor",
@@ -375,6 +376,7 @@ tf_py_test(
     python_version = "PY3",
     tfrt_enabled = True,
     deps = [
+        ":layer_utils",
         "//tensorflow/python:client_testlib",
         "//tensorflow/python:constant_op",
         "//tensorflow/python:layers",

tensorflow/python/keras/wrappers/BUILD

@@ -39,6 +39,8 @@ tf_py_test(
     deps = [
         ":wrappers",
         "//tensorflow/python:client_testlib",
+        "//tensorflow/python/keras:testing_utils",
+        "//tensorflow/python/keras/utils:np_utils",
         "//third_party/py/numpy",
     ],
 )