Automated Code Change

PiperOrigin-RevId: 650906225

Author: tensorflower-gardener

third_party/xla/xla/service/cpu/xfeed_manager.cc

@@ -22,20 +22,6 @@ namespace xla {
 namespace cpu {
 namespace runtime {
 
-void XfeedManager::Reset() {
-  infeed()->Reset();
-  outfeed()->Reset();
-}
-
-void XfeedQueueManager::Reset() {
-  absl::MutexLock l(&mu_);
-  CHECK(current_buffer_ == nullptr);
-  for (auto buffer : enqueued_buffers_) {
-    buffer->Done(ShapeUtil::MakeNil());
-  }
-  enqueued_buffers_.clear();
-}
-
 void XfeedQueueManager::EnqueueBuffersAtomically(
     absl::Span<XfeedBuffer* const> buffers) {
   absl::MutexLock l(&mu_);

third_party/xla/xla/service/cpu/xfeed_manager.h

@@ -52,13 +52,6 @@ class XfeedQueueManager {
  public:
   XfeedQueueManager(std::string queue_name) : queue_name_(queue_name) {}
 
-  // Calls the completion callback for any enqueued buffers that have
-  // not been dequeued by the runtime, and empties the
-  // queue. Reset may not be called while a runtime computation is
-  // processing a dequeued buffer. The only safe way to ensure this
-  // condition is to call Reset when no computation is taking place.
-  void Reset();
-
   // Adds a sequence of buffers to the queue atomically. buffer->Done will be
   // called when the buffer will no longer be accessed by the XfeedManager,
   // either as a result of a call to Reset or because the runtime has dequeued
@@ -108,8 +101,6 @@ class XfeedManager {
  public:
   XfeedManager() = default;
 
-  void Reset();
-
   XfeedQueueManager* infeed() { return &infeed_; }
   XfeedQueueManager* outfeed() { return &outfeed_; }