From 98edc2c7b8928d72530c402a650a56e06fd69156 Mon Sep 17 00:00:00 2001
From: David Dunleavy <ddunleavy@google.com>
Date: Thu, 30 Jan 2025 01:50:45 -0800
Subject: [PATCH] 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: 721301325
Change-Id: I167247ea821d1fae60b077dd40fdcd6ebff60d0e
---
 reverb/cc/BUILD                 | 2 +-
 reverb/cc/tensor_compression.cc | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/reverb/cc/BUILD b/reverb/cc/BUILD
index d41d689..23cd3dd 100644
--- a/reverb/cc/BUILD
+++ b/reverb/cc/BUILD
@@ -194,7 +194,7 @@ reverb_cc_library(
         "//third_party/absl/status",
         "//third_party/absl/status:statusor",
         "//third_party/absl/strings",
-        "//third_party/tensorflow/tsl/platform:status",
+        "//third_party/tensorflow/compiler/xla/tsl/platform:status",
     ] + reverb_tf_deps(),
 )
 
diff --git a/reverb/cc/tensor_compression.cc b/reverb/cc/tensor_compression.cc
index 5cab855..31be10e 100644
--- a/reverb/cc/tensor_compression.cc
+++ b/reverb/cc/tensor_compression.cc
@@ -21,11 +21,11 @@
 #include "absl/strings/str_cat.h"
 #include "reverb/cc/platform/logging.h"
 #include "reverb/cc/platform/snappy.h"
+#include "tensorflow/compiler/xla/tsl/platform/status.h"
 #include "tensorflow/core/framework/register_types.h"
 #include "tensorflow/core/framework/tensor_shape.h"
 #include "tensorflow/core/framework/types.h"
 #include "tensorflow/core/framework/types.pb.h"
-#include "tensorflow/tsl/platform/status.h"
 
 namespace deepmind {
 namespace reverb {