Method ComputeAsync

tensorflow/core/kernels/work_queue_ops.cc:310–323

  }

  void ComputeAsync(OpKernelContext* ctx,
                    AsyncOpKernel::DoneCallback done) override {
    WorkQueue* work_queue;
    OP_REQUIRES_OK(ctx,
                   LookupResource(ctx, HandleFromInput(ctx, 0), &work_queue));
    core::ScopedUnref scoped_list(work_queue);
    work_queue->Schedule(num_clients_, [this, ctx, done, work_queue]() {
      Tensor* work;
      OP_REQUIRES_OK_ASYNC(ctx, ctx->allocate_output(0, TensorShape({}), &work),
                           done);
      OP_REQUIRES_OK_ASYNC(ctx, work_queue->Take(work), done);
      done();
    });
  }

 private:
  int64 num_clients_;