litert::TensorBufferRequirements

#include <litert_tensor_buffer_requirements.h>

Represents the requirements for allocating a TensorBuffer.

Summary

This class is the C++ equivalent of LiteRtTensorBufferRequirements and is typically specified by a hardware accelerator for a given I/O tensor.

Inheritance

Inherits from: internal::Handle< LiteRtTensorBufferRequirements, LiteRtDestroyTensorBufferRequirements >

Constructors and Destructors
TensorBufferRequirements()

Public functions
Alignment() const
Expected< size_t >
BufferSize() const
Expected< size_t >
Strides() const
Expected< absl::Span< const uint32_t > >
SupportedTypes() const
Expected< std::vector< TensorBufferType > >

Public static functions
Create(absl::Span< const TensorBufferType > buffer_types, size_t buffer_size, absl::Span< const uint32_t > strides, OwnHandle owned)
CreateWithAlignment(absl::Span< const TensorBufferType > buffer_types, size_t buffer_size, size_t alignment, absl::Span< const uint32_t > strides, OwnHandle owned)
WrapCObject(LiteRtTensorBufferRequirements requirements, OwnHandle owned)

Friend classes
Join

Public functions

Alignment

Expected< size_t > Alignment() const

BufferSize

Expected< size_t > BufferSize() const

Strides

Expected< absl::Span< const uint32_t > > Strides() const

SupportedTypes

Expected< std::vector< TensorBufferType > > SupportedTypes() const

TensorBufferRequirements

 TensorBufferRequirements()=default

Public static functions

Create

Expected< TensorBufferRequirements > Create(
  absl::Span< const TensorBufferType > buffer_types,
  size_t buffer_size,
  absl::Span< const uint32_t > strides,
  OwnHandle owned
)

CreateWithAlignment

Expected< TensorBufferRequirements > CreateWithAlignment(
  absl::Span< const TensorBufferType > buffer_types,
  size_t buffer_size,
  size_t alignment,
  absl::Span< const uint32_t > strides,
  OwnHandle owned
)

WrapCObject

TensorBufferRequirements WrapCObject(
  LiteRtTensorBufferRequirements requirements,
  OwnHandle owned
)

Friend classes

Join

friend Expected< TensorBufferRequirements > Join(const TensorBufferRequirements &src1, const TensorBufferRequirements &src2)