| 505 | : AsyncOpKernel(context) {} |
| 506 | |
| 507 | void ComputeAsync(OpKernelContext* ctx, DoneCallback callback) final { |
| 508 | Barrier* barrier = nullptr; |
| 509 | OP_REQUIRES_OK_ASYNC(ctx, GetResourceFromContext(ctx, "handle", &barrier), |
| 510 | callback); |
| 511 | ComputeAsync(ctx, barrier, [callback, barrier]() { |
| 512 | barrier->Unref(); |
| 513 | callback(); |
| 514 | }); |
| 515 | } |
| 516 | |
| 517 | protected: |
| 518 | virtual void ComputeAsync(OpKernelContext* ctx, Barrier* barrier, |
nothing calls this directly
no test coverage detected