Internal visibility change

PiperOrigin-RevId: 316483590
Change-Id: Ia62c9b89846b6165be67c6cc2a3ce9f17e1b3eb3

tensorflow/BUILD

@@ -531,6 +531,7 @@ package_group(
 
 # Packages that use composite tensors or dispatch.
 # TODO(b/154762408) Remove this package group once it's no longer needed.
+# If this is modified, then copy.bara.sky must also be modified.
 package_group(name = "composite_tensor_whitelist")
 
 # Packages that use private types symbols, until they are exported.
@@ -540,6 +541,11 @@ package_group(
     packages = ["//learning/deepmind/tensorflow/replicator/..."],
 )
 
+# Packages that use StructuredTensors.
+# TODO(b/159007891) Remove this package once StructuredTensor is exported.
+# If this is modified, then copy.bara.sky must also be modified.
+package_group(name = "structured_tensor_whitelist")
+
 filegroup(
     name = "intel_binary_blob",
     data = if_mkl_ml(

tensorflow/python/ops/structured/BUILD

@@ -4,9 +4,8 @@ load("//tensorflow:tensorflow.bzl", "py_test")
 
 package(
     default_visibility = [
-        "//learning/tfx/autotfx:__subpackages__",
-        "//research/graph/convolutions/model/autotfx:__subpackages__",
         "//tensorflow:internal",
+        "//tensorflow:structured_tensor_whitelist",
     ],
     licenses = ["notice"],  # Apache 2.0
 )