| 147 | ScopedShapedBuffer::~ScopedShapedBuffer() { Deallocate(); } |
| 148 | |
| 149 | ShapedBuffer ScopedShapedBuffer::release() { |
| 150 | ShapedBuffer shaped_buffer(static_cast<ShapedBuffer&&>(*this)); |
| 151 | buffers_ = ShapeTree<se::DeviceMemoryBase>(); |
| 152 | return shaped_buffer; |
| 153 | } |
| 154 | |
| 155 | void ScopedShapedBuffer::Deallocate() { |
| 156 | // allocator_ will be null if we were moved-from. |
no outgoing calls