| 1652 | } |
| 1653 | |
| 1654 | inline void OpKernelContext::retrieve_accessed_tensors( |
| 1655 | TensorReferenceVector* out_vector) { |
| 1656 | if (params_->record_tensor_accesses) { |
| 1657 | mutex_lock l(mu_); |
| 1658 | referenced_tensors_->FreezeAndReturnReferences(out_vector); |
| 1659 | } |
| 1660 | } |
| 1661 | |
| 1662 | // no input if tensor == nullptr. |
| 1663 | inline bool OpKernelContext::has_input(int index) const { |