From 005d2962285e4093ea57cd14c5c4f029d1ce512d Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Thu, 1 Aug 2024 01:29:21 -0700
Subject: [PATCH] Automated Code Change

PiperOrigin-RevId: 658310275
---
 .../xla/xla/service/cpu/runtime/collective_thunk.cc       | 8 --------
 .../xla/xla/service/cpu/runtime/collective_thunk.h        | 3 ---
 2 files changed, 11 deletions(-)

diff --git a/third_party/xla/xla/service/cpu/runtime/collective_thunk.cc b/third_party/xla/xla/service/cpu/runtime/collective_thunk.cc
index a0cd9f4936cb33..32a452a6bcdd0d 100644
--- a/third_party/xla/xla/service/cpu/runtime/collective_thunk.cc
+++ b/third_party/xla/xla/service/cpu/runtime/collective_thunk.cc
@@ -205,10 +205,6 @@ const Shape& CollectiveThunk::source_shape(int64_t index) const {
   return op_buffers_.source_shapes[index];
 }
 
-absl::Span<const Shape> CollectiveThunk::source_shapes() const {
-  return op_buffers_.source_shapes;
-}
-
 const BufferAllocation::Slice& CollectiveThunk::destination_buffer(
     int64_t index) const {
   return op_buffers_.destination_buffers[index];
@@ -223,8 +219,4 @@ const Shape& CollectiveThunk::destination_shape(int64_t index) const {
   return op_buffers_.destination_shapes[index];
 }
 
-absl::Span<const Shape> CollectiveThunk::destination_shapes() const {
-  return op_buffers_.destination_shapes;
-}
-
 }  // namespace xla::cpu
diff --git a/third_party/xla/xla/service/cpu/runtime/collective_thunk.h b/third_party/xla/xla/service/cpu/runtime/collective_thunk.h
index 5bcf16b4e10d5c..5ae9c98844f887 100644
--- a/third_party/xla/xla/service/cpu/runtime/collective_thunk.h
+++ b/third_party/xla/xla/service/cpu/runtime/collective_thunk.h
@@ -77,7 +77,6 @@ class CollectiveThunk : public Thunk {
                   OpBuffers op_buffers, OpResources op_resources);
 
   const OpParams& op_params() const { return op_params_; }
-  const OpBuffers& op_buffers() const { return op_buffers_; }
 
   // Resolves operation's device memory from the buffers and buffer allocations.
   absl::StatusOr<OpDeviceMemory> GetOpDeviceMemory(const ExecuteParams& params);
@@ -109,13 +108,11 @@ class CollectiveThunk : public Thunk {
   absl::Span<const BufferAllocation::Slice> source_buffers() const;
 
   const Shape& source_shape(int64_t index) const;
-  absl::Span<const Shape> source_shapes() const;
 
   const BufferAllocation::Slice& destination_buffer(int64_t index) const;
   absl::Span<const BufferAllocation::Slice> destination_buffers() const;
 
   const Shape& destination_shape(int64_t index) const;
-  absl::Span<const Shape> destination_shapes() const;
 
  private:
   OpParams op_params_;