| 520 | } |
| 521 | |
| 522 | void XlaOpKernelContext::SetResourceOutput(int index, XlaResource* resource) { |
| 523 | SetOutputExpression(index, XlaExpression::Resource(resource)); |
| 524 | } |
| 525 | |
| 526 | Status XlaOpKernelContext::GetResourceInput(int index, XlaResource** resource) { |
| 527 | const XlaExpression* expression = |