Update users of TSL headers and targets to new location in XLA

Updating:
 - `env.h`
 - `env_time.h`
 - `errors.h`
 - `file_statistics.h`
 - `file_system.h`
 - `file_system_helper.h`
 - `logging.h`
 - `macros.h`
 - `status.h`
 - `status_matchers.h`
 - `status_to_from_proto.h`
 - `statusor.h`
 - `test.h`
 - `test_benchmark.h`
 - `threadpool.h`
 - `threadpool_async_executor.h`
 - `threadpool_interface.h`
 - `threadpool_options.h`
 - `types.h`

and associated targets.

PiperOrigin-RevId: 724129744

Author: ddunl

jax_tpu_embedding/sparsecore/lib/core/BUILD

@@ -28,10 +28,10 @@ cc_library(
     srcs = ["input_preprocessing_threads.cc"],
     hdrs = ["input_preprocessing_threads.h"],
     deps = [
+        "//third_party/tensorflow/compiler/xla/tsl/platform:env",
         "@com_google_absl//absl/log",
         "@com_google_absl//absl/log:check",
         "@com_google_absl//absl/strings",
-        "@tsl//tsl/platform:env",
         "@tsl//tsl/platform:platform_port",
     ],
 )
@@ -42,8 +42,8 @@ cc_test(
     env = {"JAX_PLATFORMS": "cpu"},
     deps = [
         ":input_preprocessing_threads",
+        "//third_party/tensorflow/compiler/xla/tsl/platform:env",
         "@com_google_googletest//:gtest_main",
-        "@tsl//tsl/platform:env",
     ],
 )
 

jax_tpu_embedding/sparsecore/lib/core/input_preprocessing_threads.cc

@@ -20,9 +20,9 @@
 #include "absl/log/check.h"  // from @com_google_absl
 #include "absl/log/log.h"  // from @com_google_absl
 #include "absl/strings/numbers.h"  // from @com_google_absl
+#include "third_party/tensorflow/compiler/xla/tsl/platform/env.h"
+#include "third_party/tensorflow/compiler/xla/tsl/platform/threadpool.h"
 #include "tsl/platform/cpu_info.h"  // from @tsl
-#include "tsl/platform/env.h"  // from @tsl
-#include "tsl/platform/threadpool.h"  // from @tsl
 
 namespace jax_sc_embedding {
 

jax_tpu_embedding/sparsecore/lib/core/input_preprocessing_threads.h

@@ -14,7 +14,7 @@
 #ifndef JAX_TPU_EMBEDDING_SPARSECORE_LIB_CORE_INPUT_PREPROCESSING_THREADS_H_
 #define JAX_TPU_EMBEDDING_SPARSECORE_LIB_CORE_INPUT_PREPROCESSING_THREADS_H_
 
-#include "tsl/platform/threadpool.h"  // from @tsl
+#include "third_party/tensorflow/compiler/xla/tsl/platform/threadpool.h"
 
 namespace jax_sc_embedding {
 

jax_tpu_embedding/sparsecore/lib/core/input_preprocessing_threads_test.cc

@@ -14,7 +14,7 @@
 #include "jax_tpu_embedding/sparsecore/lib/core/input_preprocessing_threads.h"
 
 #include <gtest/gtest.h>
-#include "tsl/platform/threadpool.h"  // from @tsl
+#include "third_party/tensorflow/compiler/xla/tsl/platform/threadpool.h"
 
 namespace jax_sc_embedding {
 namespace {