| 130 | } |
| 131 | |
| 132 | Status MakeOutput(const RefPtr<XRTTupleAllocation>& output, int64 index, |
| 133 | RefPtr<XRTTupleAllocation>* result) { |
| 134 | if (index == 0) { |
| 135 | *result = output; |
| 136 | } else { |
| 137 | XRTTupleAllocation* tuple; |
| 138 | TF_RETURN_IF_ERROR( |
| 139 | XRTTupleAllocation::MakeSubBuffer(output.get(), {index - 1}, &tuple, |
| 140 | /*alias_parent_allocation=*/true)); |
| 141 | result->reset(tuple); |
| 142 | } |
| 143 | return Status::OK(); |
| 144 | } |
| 145 | |
| 146 | Status PopulateOpWorkingSet(xla::Backend* backend, |
| 147 | const xrt::XRTChainedExecuteOp& op, |
no test coverage detected