#include <api_impl.h>

class BufferInstance

BufferInstance(DeviceInstance & device, iree::vm::ref< iree_hal_buffer_view_t > buffer_view)

~BufferInstance()

inline operator PJRT_Buffer *()

inline iree_hal_buffer_view_t * buffer_view()

inline DeviceInstance & device()

iree_status_t AsyncDeallocate()

iree_status_t Delete()

inline bool is_deleted()

inline bool is_on_cpu()

iree_status_t GetHostSizeInBytes(iree_host_size_t * host_size)

iree_status_t CopyToHost(void * dst, iree_host_size_t dst_size, EventInstance ** done_event)

iree_status_t AdvanceReadyFence(iree_hal_semaphore_t * semaphore, uint64_t timepoint)

iree_status_t AdvanceDoneFence(iree_hal_semaphore_t * semaphore, uint64_t timepoint)

inline iree_hal_fence_t * ready_fence()

inline iree_hal_fence_t * done_fence()

inline const int64_t * dims()

inline size_t num_dims()

std::optional< PJRT_Buffer_Type > element_type()

inline const PJRT_Buffer_MemoryLayout * layout()

static inline BufferInstance * Unwrap(PJRT_Buffer * buffer)

static void BindApi(PJRT_Api * api)

DeviceInstance & device_

iree::vm::ref< iree_hal_buffer_view_t > buffer_view_

bool is_deleted_ = false

iree::vm::ref< iree_hal_fence_t > ready_fence_

iree::vm::ref< iree_hal_fence_t > done_fence_

std::vector< int64_t > dims_

ApiMemoryLayout layout_

void ComputeLayout()