diff --git a/tensorflow_text/BUILD b/tensorflow_text/BUILD
index a6d43a6c8..5ed09fc6c 100644
--- a/tensorflow_text/BUILD
+++ b/tensorflow_text/BUILD
@@ -1249,7 +1249,21 @@ py_library(
     ],
 )
 
-# todense_py_test
+py_test(
+    name = "todense_test",
+    size = "large",
+    srcs = ["python/keras/layers/todense_test.py"],
+    python_version = "PY3",
+    shard_count = 2,
+    srcs_version = "PY3",
+    deps = [
+        ":todense_layer",
+        "@absl_py//absl/testing:parameterized",
+        #  numpy dep,
+        # tensorflow package dep,
+        # python/framework:test_lib tensorflow dep,
+    ],
+)
 
 py_library(
     name = "tokenization",
@@ -1274,7 +1288,19 @@ py_library(
     ],
 )
 
-# tokenization_layers_py_test
+py_test(
+    name = "tokenization_layers_test",
+    size = "large",
+    srcs = ["python/keras/layers/tokenization_layers_test.py"],
+    python_version = "PY3",
+    shard_count = 20,
+    deps = [
+        ":tokenization_layers",
+        "@absl_py//absl/testing:parameterized",
+        #  numpy dep,
+        # tensorflow package dep,
+    ],
+)
 
 py_tf_text_library(
     name = "trimmer_ops",