Searched refs:EnqueueOp (Results 1 – 4 of 4) sorted by relevance
27 REGISTER_KERNEL_BUILDER(Name("QueueEnqueue").Device(DEVICE_CPU), EnqueueOp);28 REGISTER_KERNEL_BUILDER(Name("QueueEnqueueV2").Device(DEVICE_CPU), EnqueueOp);61 EnqueueOp);
94 class EnqueueOp : public QueueAccessOpKernel {96 explicit EnqueueOp(OpKernelConstruction* context);103 TF_DISALLOW_COPY_AND_ASSIGN(EnqueueOp);
85 EnqueueOp::EnqueueOp(OpKernelConstruction* context) in EnqueueOp() function in tensorflow::EnqueueOp88 void EnqueueOp::ComputeAsync(OpKernelContext* ctx, QueueInterface* queue, in ComputeAsync()
167 using EnqueueOp = typename EnqueueOpWrapper<XlaInputs, N>::type;178 void Run(EnqueueOp enqueue_op, EvaluateOp evaluate_op) {189 void Run(EnqueueOp enqueue_op, EvaluateOp evaluate_op,