sparsity

AirDimensionMetaData()

Metadata for a dimension in a sparse tensor.

format: int = Field(0, description='Format of the dimension')

Format of the dimension.

dense_size: int = Field(0, description='Dense size of the dimension')

Dense size of the dimension.

array_segments_type: int = Field(0, description='Array segments type')

Array segments type.

array_segments: list[int] | None = Field(None, description='Array segments')

Array segments.

array_indices_type: int = Field(0, description='Array indices type')

Array indices type.

array_indices: list[int] | None = Field(None, description='Array indices')

Array indices.

AirSparsityParameters()

Parameters for sparsity in a tensor.

traversal_order: list[int] | None = Field(None, description='Traversal order')

Traversal order.

block_map: list[int] | None = Field(None, description='Block map')

Block map.

dim_meta_data: list[AirDimensionMetaData] | None = Field(None, description='Dimension metadata')

Dimension metadata.