Home
last modified time | relevance | path

Searched refs:QueueInterface (Results 1 – 16 of 16) 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()
108 QueueInterface* queue; in ComputeWithReader()
Dfifo_queue.h82 Status CreateResource(QueueInterface** ret) override
Dqueue_base.h40 class QueueBase : public QueueInterface {
Dfifo_queue.cc378 Status FIFOQueueOp::CreateResource(QueueInterface** ret) { in CreateResource()
Drandom_shuffle_queue_op.cc495 Status CreateResource(QueueInterface** ret) override in CreateResource()
/external/tensorflow/tensorflow/core/framework/
Dreader_interface.h29 class QueueInterface; variable
51 virtual void Read(QueueInterface* queue, string* key, string* value,
62 virtual int64 ReadUpTo(const int64 num_records, QueueInterface* queue,
Dreader_base.h108 virtual string GetNextWorkLocked(QueueInterface* queue,
113 void Read(QueueInterface* queue, string* key, string* value,
118 int64 ReadUpTo(const int64 num_records, QueueInterface* queue,
Dreader_base.cc78 int64 ReaderBase::ReadUpTo(const int64 num_records, QueueInterface* queue, in ReadUpTo()
153 void ReaderBase::Read(QueueInterface* queue, string* key, string* value, in Read()
200 string ReaderBase::GetNextWorkLocked(QueueInterface* queue, in GetNextWorkLocked()
205 context, [this, context, &n, &work](const QueueInterface::Tuple& tuple) { in GetNextWorkLocked()
Dqueue_interface.h31 class QueueInterface : public ResourceBase {
97 virtual ~QueueInterface() {} in ~QueueInterface()
/external/tensorflow/tensorflow/core/common_runtime/sycl/
Dsycl_device.h31 std::vector<Eigen::QueueInterface*> m_queue_interface_; // owned
116 m_queue_interface_.push_back(new Eigen::QueueInterface(d)); in AddDevice()
129 Eigen::QueueInterface* GetQueueInterface(size_t i = 0) const {
166 Eigen::QueueInterface* queue_ptr = GetQueueInterface(device_id);
Dsycl_allocator.h32 SYCLAllocator(Eigen::QueueInterface* queue);
Dsycl_allocator.cc22 SYCLAllocator::SYCLAllocator(Eigen::QueueInterface* queue) in SYCLAllocator()