Internal infra change

PiperOrigin-RevId: 372261376
Change-Id: I96eb1497a732bc813faf2a38bbf3dd1ff213fd62
diff --git a/tensorflow/lite/python/BUILD b/tensorflow/lite/python/BUILD
index 8fb3540..b56f7ba 100644
--- a/tensorflow/lite/python/BUILD
+++ b/tensorflow/lite/python/BUILD
@@ -1,6 +1,7 @@
 load("//tensorflow:tensorflow.bzl", "pytype_strict_library")
 load("@flatbuffers//:build_defs.bzl", "flatbuffer_py_library")
 load("//tensorflow:tensorflow.bzl", "get_compatible_with_portable", "if_portable")
+load("//tensorflow/lite:special_rules.bzl", "internal_visibility_allowlist")
 
 package(
     default_visibility = [
@@ -270,9 +271,7 @@
     name = "util",
     srcs = ["util.py"],
     srcs_version = "PY3",
-    visibility = [
-        "//tensorflow/lite:__subpackages__",
-    ],
+    visibility = internal_visibility_allowlist(),
     deps = [
         ":op_hint",
         ":schema_py",