[pull] master from tensorflow:master

.github/workflows/check_license.yml

@@ -0,0 +1,56 @@
+# Copyright 2026 The TensorFlow Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ============================================================================
+name: License Check
+
+permissions: {}
+
+on:
+  pull_request:
+  push:
+    branches:
+      - main
+
+jobs:
+  license-check:
+    runs-on: ubuntu-latest
+    defaults:
+      run:
+        shell: bash
+    timeout-minutes: 10
+    steps:
+      - name: "Checking out repository"
+        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd # v6.0.2
+        with:
+          persist-credentials: false
+      - name: "Set up Go"
+        uses: actions/setup-go@4a3601121dd01d1626a1e23e37211e3254c1c06c # v6.4.0
+        with:
+          go-version: 'stable'
+      - name: "Run license check"
+        uses: google-ml-infra/actions/ci_check_license@432d36c6ac62d76fa99b3679758ccfbc10140fb6
+        with:
+          copyright-holder: 'The TensorFlow Authors'
+          license-type: 'apache'
+          exclude-paths: 'third_party'
+          exclude-files: >-
+            tensorflow_issue_template.yaml
+            BUILD
+            _redirects.yaml
+            cloudbuild.yaml
+            _index.yaml
+            _book.yaml
+            _toc.yaml
+            _doxygen.yaml
+          exclude-extensions: 'gradle'

tensorflow/compiler/jit/BUILD

@@ -12,7 +12,6 @@ load("//tensorflow/compiler/jit:package_groups.bzl", "legacy_jit_users_package_g
 load("//tensorflow/core/platform:build_config.bzl", "tf_additional_all_protos", "tf_proto_library")
 load(
     "//tensorflow/core/platform:build_config_root.bzl",
-    "if_static",
     "tf_cuda_tests_tags",
 )
 load("//tensorflow/core/platform:rules_cc.bzl", "cc_library")
@@ -134,7 +133,6 @@ cc_library(
     deps = if_libtpu([
         "//tensorflow/core/tpu/graph_rewrite:tpu_rewrite_pass_registration",
         "//tensorflow/core/tpu/graph_rewrite:configure_tpu_embedding_rewrite_registration",
-        "@xla//xla/stream_executor/tpu:tpu_transfer_manager",
     ]),
     alwayslink = 1,
 )
@@ -202,42 +200,15 @@ cc_library(
 cc_library(
     name = "xla_tpu_device",
     srcs = ["xla_tpu_device.cc"],
-    hdrs = ["xla_tpu_device.h"],
     visibility = [":friends"],
     deps = [
         ":xla_device",
-        ":xla_device_context",
-        ":xla_kernel_creator",  # buildcleaner: keep
-        "//tensorflow/compiler/jit/kernels:xla_ops",
-        "//tensorflow/compiler/tf2xla:common",
-        "//tensorflow/compiler/tf2xla:layout_util",
-        "//tensorflow/compiler/tf2xla:tf2xla_util",
-        "//tensorflow/compiler/tf2xla:xla_helpers",
-        "//tensorflow/compiler/tf2xla:xla_op_registry",
+        "//tensorflow/compiler/jit/kernels:xla_ops_no_jit_rewrite_registration",
+        "//tensorflow/core:core_cpu_base",
         "//tensorflow/core:framework_internal",
-        "//tensorflow/core:lib_proto_parsing",
-        "//tensorflow/core:protos_all_cc",
-        "//tensorflow/core:session_options",
-        "//tensorflow/core/common_runtime:device",
-        "//tensorflow/core/common_runtime:device_factory",
-        "//tensorflow/core/common_runtime:dma_helper",
-        "//tensorflow/core/platform:status",
+        "//tensorflow/core/framework:kernel_def_proto_cc",
         "//tensorflow/core/tpu:tpu_defs",
-        "//tensorflow/core/tpu:tpu_node_device_util",
-        "//tensorflow/core/tpu:virtual_device",
-        "@com_google_absl//absl/types:optional",
-        "@tsl//tsl/platform:statusor",
-        "@xla//xla/stream_executor/tpu:tpu_executor_base",
-        "@xla//xla/stream_executor/tpu:tpu_node_context",
-        "@xla//xla/stream_executor/tpu:tpu_platform_interface",
-        "@xla//xla/stream_executor/tpu:tpu_stream_interface",
-        "@xla//xla/tpu:c_api_conversions",
-        "@xla//xla/tpu:status_helper",
-        "@xla//xla/tpu:tpu_api",
-    ] + if_static([
-        "//tensorflow/core/common_runtime:copy_tensor",
-        ":jit_compilation_passes",
-    ]),
+    ],
     alwayslink = 1,
 )