Method Compute

tensorflow/core/kernels/work_queue_ops.cc:274–281

  explicit WorkQueueSizeOp(OpKernelConstruction* ctx) : OpKernel(ctx) {}

  void Compute(OpKernelContext* ctx) override {
    WorkQueue* work_queue;
    OP_REQUIRES_OK(ctx,
                   LookupResource(ctx, HandleFromInput(ctx, 0), &work_queue));
    Tensor* size;
    OP_REQUIRES_OK(ctx, ctx->allocate_output(0, TensorShape({}), &size));
    OP_REQUIRES_OK(ctx, work_queue->GetSize(size));
  }
};

REGISTER_KERNEL_BUILDER(Name("WorkQueueSize").Device(DEVICE_CPU),