litert::TensorBuffer

Summary

Inheritance

Inherits from: litert::internal::Handle< LiteRtTensorBuffer, LiteRtDestroyTensorBuffer >

Constructors and Destructors

TensorBuffer()
TensorBuffer(LiteRtTensorBuffer tensor_buffer, OwnHandle owned)

Public functions

BufferType() const
Expected< LiteRtTensorBufferType >
ClearEvent()
Expected< void >
Duplicate() const
GetAhwb() const
litert::Expected< AHardwareBuffer * >
GetDmaBuf() const
GetEvent() const
GetGlBuffer() const
GetGlTexture() const
GetOpenClMemory() const
Expected< cl_mem >
HasEvent() const
bool
IsOpenClMemory() const
bool
Lock()
Expected< void * >
Offset() const
Expected< size_t >
PackedSize() const
Expected< size_t >
Read(absl::Span< T > data)
Expected< void >
SetEvent(Event && event)
Expected< void >
SetLiteRtEvent(LiteRtEvent & litert_event)
Expected< void >
Size() const
Expected< size_t >
TensorType() const
Unlock()
Expected< void >
Write(absl::Span< const T > data)
Expected< void >

Public static functions

CreateFromAhwb(const RankedTensorType & tensor_type, AHardwareBuffer *ahwb, size_t ahwb_offset)
CreateFromGlBuffer(const RankedTensorType & tensor_type, LiteRtGLenum target, LiteRtGLuint id, size_t size_bytes, size_t offset)
CreateFromGlTexture(const RankedTensorType & tensor_type, LiteRtGLenum target, LiteRtGLuint id, LiteRtGLenum format, size_t size_bytes, LiteRtGLint layer)
CreateFromHostMemory(const RankedTensorType & tensor_type, void *host_mem_addr, size_t buffer_size)
CreateManaged(LiteRtTensorBufferType buffer_type, const RankedTensorType & tensor_type, size_t buffer_size)

Structs

litert::TensorBuffer::DmaBuf
litert::TensorBuffer::GlBuffer
litert::TensorBuffer::GlTexture

Public functions

BufferType

Expected< LiteRtTensorBufferType > BufferType() const 

ClearEvent

Expected< void > ClearEvent()

Duplicate

Expected< TensorBuffer > Duplicate() const 

GetAhwb

litert::Expected< AHardwareBuffer * > GetAhwb() const 

GetDmaBuf

litert::Expected< DmaBuf > GetDmaBuf() const 

GetEvent

Expected< Event > GetEvent() const 

GetGlBuffer

Expected< GlBuffer > GetGlBuffer() const 

GetGlTexture

Expected< GlTexture > GetGlTexture() const 

GetOpenClMemory

Expected< cl_mem > GetOpenClMemory() const 

HasEvent

bool HasEvent() const 

IsOpenClMemory

bool IsOpenClMemory() const 

Lock

Expected< void * > Lock()

Offset

Expected< size_t > Offset() const 

PackedSize

Expected< size_t > PackedSize() const 

Read

Expected< void > Read(
  absl::Span< T > data
)

SetEvent

Expected< void > SetEvent(
  Event && event
)

SetLiteRtEvent

Expected< void > SetLiteRtEvent(
  LiteRtEvent & litert_event
)

Size

Expected< size_t > Size() const 

TensorBuffer

 TensorBuffer()=default

TensorBuffer

 TensorBuffer(
  LiteRtTensorBuffer tensor_buffer,
  OwnHandle owned
)

TensorType

Expected< RankedTensorType > TensorType() const 

Unlock

Expected< void > Unlock()

Write

Expected< void > Write(
  absl::Span< const T > data
)

Public static functions

CreateFromAhwb

Expected< TensorBuffer > CreateFromAhwb(
  const RankedTensorType & tensor_type,
  AHardwareBuffer *ahwb,
  size_t ahwb_offset
)

CreateFromGlBuffer

Expected< TensorBuffer > CreateFromGlBuffer(
  const RankedTensorType & tensor_type,
  LiteRtGLenum target,
  LiteRtGLuint id,
  size_t size_bytes,
  size_t offset
)

CreateFromGlTexture

Expected< TensorBuffer > CreateFromGlTexture(
  const RankedTensorType & tensor_type,
  LiteRtGLenum target,
  LiteRtGLuint id,
  LiteRtGLenum format,
  size_t size_bytes,
  LiteRtGLint layer
)

CreateFromHostMemory

Expected< TensorBuffer > CreateFromHostMemory(
  const RankedTensorType & tensor_type,
  void *host_mem_addr,
  size_t buffer_size
)

CreateManaged

Expected< TensorBuffer > CreateManaged(
  LiteRtTensorBufferType buffer_type,
  const RankedTensorType & tensor_type,
  size_t buffer_size
)