From a3fbe22e4c76dbd77d5c8a2707da6ed78e76dbb7 Mon Sep 17 00:00:00 2001 From: David Dunleavy Date: Thu, 17 Nov 2022 22:36:44 -0800 Subject: [PATCH] Remove deps on Tensorflow in XLA PiperOrigin-RevId: 489390169 --- tensorflow/compiler/xla/python/tpu_driver/BUILD | 1 - tensorflow/compiler/xla/service/BUILD | 3 +-- .../compiler/xla/translate/mhlo_to_lhlo_with_xla/tests/BUILD | 2 +- 3 files changed, 2 insertions(+), 4 deletions(-) diff --git a/tensorflow/compiler/xla/python/tpu_driver/BUILD b/tensorflow/compiler/xla/python/tpu_driver/BUILD index 440aba3dc89..6952c57b8e1 100644 --- a/tensorflow/compiler/xla/python/tpu_driver/BUILD +++ b/tensorflow/compiler/xla/python/tpu_driver/BUILD @@ -134,7 +134,6 @@ cc_library( "//tensorflow/tsl/platform:env", "//tensorflow/tsl/platform:errors", "//tensorflow/tsl/platform:logging", - "//tensorflow/core:protos_all_cc", ] + tsl_grpc_cc_dependencies() + external_deps(), alwayslink = 1, ) diff --git a/tensorflow/compiler/xla/service/BUILD b/tensorflow/compiler/xla/service/BUILD index 38403a4e26b..d1f35b53061 100644 --- a/tensorflow/compiler/xla/service/BUILD +++ b/tensorflow/compiler/xla/service/BUILD @@ -7003,9 +7003,8 @@ tf_cc_binary( "//tensorflow/compiler/xla/pjrt:mlir_to_hlo", "//tensorflow/compiler/xla/service/cpu:cpu_compiler", "//tensorflow/compiler/xla/service/cpu:cpu_executable", - "//tensorflow/core:framework_internal", - "//tensorflow/core:lib_proto_parsing", "//tensorflow/tsl/platform:env", + "//tensorflow/tsl/platform:platform_port", "//tensorflow/tsl/util:command_line_flags", "@llvm-project//mlir:ArithDialect", "@llvm-project//mlir:FuncDialect", diff --git a/tensorflow/compiler/xla/translate/mhlo_to_lhlo_with_xla/tests/BUILD b/tensorflow/compiler/xla/translate/mhlo_to_lhlo_with_xla/tests/BUILD index aab0814c706..bed57ce10ff 100644 --- a/tensorflow/compiler/xla/translate/mhlo_to_lhlo_with_xla/tests/BUILD +++ b/tensorflow/compiler/xla/translate/mhlo_to_lhlo_with_xla/tests/BUILD @@ -1,7 +1,7 @@ load("//tensorflow/tsl:tsl.default.bzl", "filegroup") load("//tensorflow/compiler/mlir:glob_lit_test.bzl", "glob_lit_tests") load( - "//tensorflow/core/platform:build_config_root.bzl", + "//tensorflow/tsl/platform:build_config_root.bzl", "tf_cuda_tests_tags", )