| 86 | class DelayOp : public AsyncOpKernel { |
| 87 | public: |
| 88 | explicit DelayOp(OpKernelConstruction* ctx) : AsyncOpKernel(ctx) { |
| 89 | OP_REQUIRES_OK(ctx, ctx->GetAttr("micros", µs_)); |
| 90 | } |
| 91 | |
| 92 | void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { |
| 93 | ctx->set_output(0, ctx->input(0)); |