Internal change to build rules

PiperOrigin-RevId: 308727626
Change-Id: I9d092022d1068c314bc7671e43cb44f17534025c

tensorflow/BUILD

@@ -525,6 +525,10 @@ package_group(
 
 package_group(name = "ndarray_tensor_allow_list")
 
+# Packages that use composite tensors or dispatch.
+# TODO(b/154762408) Remove this package group once it's no longer needed.
+package_group(name = "composite_tensor_whitelist")
+
 filegroup(
     name = "intel_binary_blob",
     data = if_mkl_ml(

tensorflow/python/BUILD

@@ -1772,6 +1772,7 @@ py_library(
     name = "composite_tensor",
     srcs = ["framework/composite_tensor.py"],
     srcs_version = "PY2AND3",
+    visibility = visibility + ["//tensorflow:composite_tensor_whitelist"],
     deps = [
         ":dtypes",
         ":tensor_util",
@@ -1845,6 +1846,7 @@ py_library(
     name = "type_spec",
     srcs = ["framework/type_spec.py"],
     srcs_version = "PY2AND3",
+    visibility = visibility + ["//tensorflow:composite_tensor_whitelist"],
     deps = [
         ":dtypes",
         ":tensor_shape",
@@ -5528,6 +5530,7 @@ py_library(
         "//tensorflow:__pkg__",
         "//third_party/py/tensorflow_core:__subpackages__",
         "//third_party/py/tf_agents:__subpackages__",
+        "//tensorflow:composite_tensor_whitelist",
     ],
     deps = [
         ":tf_decorator",