An op kernel can access the tensor store of the run it belongs to.
| 1152 | |
| 1153 | // An op kernel can access the tensor store of the run it belongs to. |
| 1154 | TensorStore* tensor_store() const { return params_->tensor_store; } |
| 1155 | |
| 1156 | // Function call support. |
| 1157 | // |