| 1667 | } |
| 1668 | |
| 1669 | inline Tensor* OpKernelContext::input_ref(int index) { |
| 1670 | DCHECK_GE(index, 0); |
| 1671 | DCHECK_LT(index, num_inputs()); |
| 1672 | DCHECK(input_is_ref(index)); |
| 1673 | return (*params_->inputs)[index].tensor; |
| 1674 | } |
| 1675 | |
| 1676 | inline mutex* OpKernelContext::input_ref_mutex(int index) { |
| 1677 | DCHECK_GE(index, 0); |