| 333 | bool XRTTupleAllocation::IsPinned() const { return pin_count_ != 0; } |
| 334 | |
| 335 | void XRTTupleAllocation::DiscardAllocation( |
| 336 | const xla::ShapeIndex& buffer_index) { |
| 337 | buffers_.element(buffer_index)->DiscardAllocation(); |
| 338 | } |
| 339 | |
| 340 | const xla::Shape& XRTTupleAllocation::on_host_shape() const { |
| 341 | return on_host_shape_; |