| 343 | } |
| 344 | |
| 345 | void OpKernelContext::really_record_tensor_reference(const Tensor& tensor) { |
| 346 | mutex_lock l(mu_); |
| 347 | // Keep a reference to the underlying memory around. |
| 348 | referenced_tensors_->Add(tensor); |
| 349 | } |
| 350 | |
| 351 | Status OpKernelContext::input(StringPiece name, const Tensor** tensor) { |
| 352 | int start, stop; |