| 117 | } |
| 118 | |
| 119 | ScopedShapedBuffer::ScopedShapedBuffer(Shape on_host_shape, |
| 120 | Shape on_device_shape, |
| 121 | se::DeviceMemoryAllocator* allocator, |
| 122 | int device_ordinal) |
| 123 | : ShapedBuffer(std::move(on_host_shape), std::move(on_device_shape), |
| 124 | allocator->platform(), device_ordinal), |
| 125 | allocator_(allocator) {} |
| 126 | |
| 127 | ScopedShapedBuffer::ScopedShapedBuffer(ShapedBuffer shaped_buffer, |
| 128 | se::DeviceMemoryAllocator* allocator) |