| 524 | } |
| 525 | |
| 526 | Status XlaOpKernelContext::GetResourceInput(int index, XlaResource** resource) { |
| 527 | const XlaExpression* expression = |
| 528 | CastExpressionFromTensor(context_->input(index)); |
| 529 | TF_RET_CHECK(expression->resource() != nullptr); |
| 530 | *resource = expression->resource(); |
| 531 | return Status::OK(); |
| 532 | } |
| 533 | |
| 534 | namespace { |
| 535 |