diff --git a/tensorflow/python/profiler/internal/BUILD b/tensorflow/python/profiler/internal/BUILD index 077e1710660d88..07ede25fcdd572 100644 --- a/tensorflow/python/profiler/internal/BUILD +++ b/tensorflow/python/profiler/internal/BUILD @@ -200,6 +200,15 @@ tsl_pybind_extension( "//tensorflow/python/profiler:__pkg__", ]), deps = [ + "@local_xla//xla/tsl/platform/default:resource", + "@local_tsl//tsl/platform:random", + "@local_xla//xla/tsl/lib/monitoring:collection_registry", + "//tensorflow/core:framework_internal_impl", + "@local_xla//xla/tsl/lib/histogram:histogram", + "@local_xla//xla/tsl/lib/monitoring:sampler", + "//tensorflow/core/platform:platform_strings", + "//tensorflow/core/lib/core:arena", + "//tensorflow/core/common_runtime/gpu:gpu_id_impl", # weekly-sync:24-12-10 "//tensorflow/core:protos_all_cc_impl", "//tensorflow/core/framework:attr_value_proto_cc_impl", "//tensorflow/core/framework:op", @@ -209,6 +218,10 @@ tsl_pybind_extension( "//tensorflow/core/profiler/convert:xplane_to_tools_data", "//tensorflow/python/lib/core:py_exception_registry", "//tensorflow/python/lib/core:pybind11_status", + "@com_google_absl//absl/container:flat_hash_map", + "@com_google_absl//absl/log", + "@com_google_absl//absl/status", + "@com_google_absl//absl/status:statusor", "@com_google_protobuf//:protobuf", "@local_tsl//tsl/platform:env_impl", "@local_tsl//tsl/profiler/protobuf:profiler_analysis_proto_cc_impl", @@ -256,4 +269,4 @@ tsl_pybind_extension( "@local_tsl//tsl/profiler/lib:profiler_factory_impl", "@local_tsl//tsl/profiler/lib:profiler_session_impl", ]), -) +) \ No newline at end of file