From 13322e711c6e99966856335759c04d0771e60eb7 Mon Sep 17 00:00:00 2001 From: Yin Zhang Date: Fri, 10 Jan 2025 20:55:04 -0800 Subject: [PATCH] Internal relative changes only PiperOrigin-RevId: 714327465 --- tensorflow/core/profiler/utils/BUILD | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tensorflow/core/profiler/utils/BUILD b/tensorflow/core/profiler/utils/BUILD index 460d2cb046e0ba..cf0dd5728ce3c9 100644 --- a/tensorflow/core/profiler/utils/BUILD +++ b/tensorflow/core/profiler/utils/BUILD @@ -474,6 +474,10 @@ tf_cuda_library( cc_library( name = "hlo_module_utils", hdrs = ["hlo_module_utils.h"], + visibility = [ + ":friends", + # copybara:uncomment "//tensorflow/compiler/mlir/lite/experimental/google/tooling/google:__subpackages__", + ], deps = [ "@com_google_absl//absl/strings", "@local_xla//xla/hlo/ir:hlo",