Adds a xla_bzl_visibility macro for declaring the visibility of a bzl…

… file.

PiperOrigin-RevId: 728376113

xla/tsl/framework/contraction/build_defs.bzl

@@ -1,5 +1,16 @@
 """Defines the cc_binary_disable_onednn build rule to disable oneDNN."""
 
+load("//xla/tsl:tsl.bzl", "xla_bzl_visibility")
+
+visibility(xla_bzl_visibility([
+    "platforms/xla/...",
+    "third_party/tensorflow/...",
+    # The following are legacy users of this .bzl file. DO NOT ADD NEW USERS.
+    # go/keep-sorted start
+    "tech/internal/env/extension",
+    # go/keep-sorted end
+]))
+
 def _disable_onednn_transition_impl(settings, attr):
     _ignore = (settings, attr)  # @unused
     return {"//xla/tsl/framework/contraction:disable_onednn_contraction_kernel": True}

xla/tsl/tsl.bzl

@@ -122,6 +122,19 @@ def internal_visibility(internal_targets):
     """
     return if_google(internal_targets, ["//visibility:public"])
 
+def _make_abs_targets(targets):
+    return ["//" + target for target in targets]
+
+def xla_bzl_visibility(targets):
+    """To be used in a .bzl file's visibility declaration:
+
+    visibility(xla_bzl_visibility(targets))
+
+    expands to visibility(targets) inside Google, but visibility("public") in OSS.
+    """
+
+    return if_google(_make_abs_targets(targets), "public")
+
 # TODO(jakeharmon): Use this to replace if_static
 # TODO(b/356020232): remove completely after migration is done
 def if_tsl_link_protobuf(if_true, if_false = []):