Home
last modified time | relevance | path

Searched refs:QueueInterface (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dqueue_op.h34 class QueueOp : public ResourceOpKernel<QueueInterface> {
46 Status VerifyResource(QueueInterface* queue) override;
55 Status CreateTypedQueue(TypedQueue* queue, QueueInterface** ret) { in CreateTypedQueue()
73 virtual void ComputeAsync(OpKernelContext* ctx, QueueInterface* queue,
99 void ComputeAsync(OpKernelContext* ctx, QueueInterface* queue,
124 void ComputeAsync(OpKernelContext* ctx, QueueInterface* queue,
145 void ComputeAsync(OpKernelContext* ctx, QueueInterface* queue,
170 void ComputeAsync(OpKernelContext* ctx, QueueInterface* queue,
213 void ComputeAsync(OpKernelContext* ctx, QueueInterface* queue,
231 void ComputeAsync(OpKernelContext* ctx, QueueInterface* queue,
[all …]
Dqueue_op.cc38 ResourceOpKernel<QueueInterface>::Compute(context); in Compute()
45 Status QueueOp::VerifyResource(QueueInterface* queue) { in VerifyResource()
54 QueueInterface* queue; in ComputeAsync()
88 void EnqueueOp::ComputeAsync(OpKernelContext* ctx, QueueInterface* queue, in ComputeAsync()
101 QueueInterface::Tuple tuple; in ComputeAsync()
129 void EnqueueManyOp::ComputeAsync(OpKernelContext* ctx, QueueInterface* queue, in ComputeAsync()
142 QueueInterface::Tuple tuple; in ComputeAsync()
166 void DequeueOp::ComputeAsync(OpKernelContext* ctx, QueueInterface* queue, in ComputeAsync()
178 queue->TryDequeue(ctx, [ctx, callback](const QueueInterface::Tuple& tuple) { in ComputeAsync()
209 void DequeueManyOp::ComputeAsync(OpKernelContext* ctx, QueueInterface* queue, in ComputeAsync()
[all …]
Dpriority_queue_op.cc53 Status CreateResource(QueueInterface** ret) override in CreateResource()
Dpadding_fifo_queue_op.cc56 Status CreateResource(QueueInterface** ret) override in CreateResource()
Dreader_ops.cc82 QueueInterface* queue; in ComputeWithReader()
111 QueueInterface* queue; in ComputeWithReader()
Dfifo_queue.h82 Status CreateResource(QueueInterface** ret) override
Dqueue_base.h36 class QueueBase : public QueueInterface {
Dfifo_queue.cc378 Status FIFOQueueOp::CreateResource(QueueInterface** ret) { in CreateResource()
Drandom_shuffle_queue_op.cc496 Status CreateResource(QueueInterface** ret) override in CreateResource()
/external/tensorflow/tensorflow/core/framework/
Dreader_interface.h29 class QueueInterface; variable
51 virtual void Read(QueueInterface* queue, tstring* key, tstring* value,
62 virtual int64 ReadUpTo(const int64 num_records, QueueInterface* queue,
Dreader_base.h108 virtual string GetNextWorkLocked(QueueInterface* queue,
113 void Read(QueueInterface* queue, tstring* key, tstring* value,
118 int64 ReadUpTo(const int64 num_records, QueueInterface* queue,
Dqueue_interface.h31 class QueueInterface : public ResourceBase {
97 virtual ~QueueInterface() {} in ~QueueInterface()
Dreader_base.cc78 int64 ReaderBase::ReadUpTo(const int64 num_records, QueueInterface* queue, in ReadUpTo()
153 void ReaderBase::Read(QueueInterface* queue, tstring* key, tstring* value, in Read()
200 string ReaderBase::GetNextWorkLocked(QueueInterface* queue, in GetNextWorkLocked()
205 context, [context, &n, &work](const QueueInterface::Tuple& tuple) { in GetNextWorkLocked()