Searched refs:QueueSizeOp (Results 1 – 4 of 4) sorted by relevance
245 class QueueSizeOp : public QueueOpKernel {247 explicit QueueSizeOp(OpKernelConstruction* context);254 TF_DISALLOW_COPY_AND_ASSIGN(QueueSizeOp);
51 REGISTER_KERNEL_BUILDER(Name("QueueSize").Device(DEVICE_CPU), QueueSizeOp);52 REGISTER_KERNEL_BUILDER(Name("QueueSizeV2").Device(DEVICE_CPU), QueueSizeOp);
344 QueueSizeOp::QueueSizeOp(OpKernelConstruction* context) in QueueSizeOp() function in tensorflow::QueueSizeOp347 void QueueSizeOp::ComputeAsync(OpKernelContext* ctx, QueueInterface* queue, in ComputeAsync()
202 QueueSizeOp); \