| 1687 | } |
| 1688 | |
| 1689 | inline Tensor* OpKernelContext::mutable_output(int index) { |
| 1690 | DCHECK_GE(index, 0); |
| 1691 | DCHECK_LT(index, num_outputs()); |
| 1692 | // No need to record_tensor_reference since the output must already |
| 1693 | // have been set by a call that did so. |
| 1694 | return outputs_[index].tensor; |
| 1695 | } |
| 1696 | |
| 1697 | inline TensorValue OpKernelContext::release_output(int index) { |
| 1698 | DCHECK_GE(index, 0); |
no outgoing calls