| 54 | } |
| 55 | |
| 56 | void Compute(OpKernelContext* context) override { |
| 57 | // We always return the input ref. |
| 58 | context->forward_ref_input_to_ref_output(0, 0); |
| 59 | |
| 60 | if (use_exclusive_lock_) { |
| 61 | mutex_lock l(*context->input_ref_mutex(0)); |
| 62 | DoUpdate(context); |
| 63 | } else { |
| 64 | DoUpdate(context); |
| 65 | } |
| 66 | } |
| 67 | |
| 68 | private: |
| 69 | void DoUpdate(OpKernelContext* context) { |
nothing calls this directly
no test coverage detected