HomeSort by relevance Sort by last modified time
    Searched refs:XfeedBuffer (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
xfeed_manager.h 38 class XfeedBuffer {
40 virtual ~XfeedBuffer() = default;
46 // expecting / producing with respect to this XfeedBuffer. E.g. this will
67 void EnqueueBuffersAtomically(absl::Span<XfeedBuffer* const> buffers);
74 XfeedBuffer* BlockingDequeueBuffer();
97 // XfeedBuffer* queue contents are not owned, but buffer->Done must
99 std::deque<XfeedBuffer*> enqueued_buffers_;
103 XfeedBuffer* current_buffer_ = nullptr;
xfeed_manager.cc 40 absl::Span<XfeedBuffer* const> buffers) {
43 for (XfeedBuffer* b : buffers) {
57 XfeedBuffer* XfeedQueueManager::BlockingDequeueBuffer() {
cpu_transfer_manager.cc 43 class CpuInfeedBuffer : public cpu::runtime::XfeedBuffer {
63 class CpuOutfeedBuffer : public cpu::runtime::XfeedBuffer {
113 std::vector<cpu::runtime::XfeedBuffer*> buffers;
116 for (cpu::runtime::XfeedBuffer* b : buffers) {
125 cpu::runtime::XfeedBuffer * buffer,
142 TF_ASSIGN_OR_RETURN(cpu::runtime::XfeedBuffer * buffer,
152 StatusOr<cpu::runtime::XfeedBuffer*>
264 std::vector<cpu::runtime::XfeedBuffer*> buffer_pointers;
cpu_transfer_manager.h 53 StatusOr<cpu::runtime::XfeedBuffer*> TransferBufferToInfeedInternal(
cpu_runtime.cc 121 xla::cpu::runtime::XfeedBuffer* buffer =
164 xla::cpu::runtime::XfeedBuffer* buffer =
xfeed_manager_test.cc 33 class TestInfeedBuffer : public cpu::runtime::XfeedBuffer {

Completed in 500 milliseconds