Home
last modified time | relevance | path

Searched refs:current_buffer_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dxfeed_manager.cc32 CHECK(current_buffer_ == nullptr); in Reset()
64 CHECK(current_buffer_ == nullptr); in BlockingDequeueBuffer()
65 current_buffer_ = enqueued_buffers_.front(); in BlockingDequeueBuffer()
67 return current_buffer_; in BlockingDequeueBuffer()
76 CHECK(current_buffer_ != nullptr); in ReleaseCurrentBuffer()
77 CHECK_EQ(length, current_buffer_->length()); in ReleaseCurrentBuffer()
78 CHECK_EQ(data, current_buffer_->data()); in ReleaseCurrentBuffer()
79 current_buffer_->Done(std::move(shape)); in ReleaseCurrentBuffer()
80 current_buffer_ = nullptr; in ReleaseCurrentBuffer()
Dxfeed_manager.h103 XfeedBuffer* current_buffer_ = nullptr; variable