mlir.dialects._sparse_tensor_enum_gen

Attributes

_ods_ir

Classes

CrdTransDirectionKind	sparse tensor coordinate translation direction
SparseTensorSortKind	sparse tensor sort algorithm
StorageSpecifierKind	sparse tensor storage specifier kind

Functions

_sparsetensorcrdtransdirectionenum(x, context)
_sparsetensorsortkindenum(x, context)
_sparsetensorstoragespecifierkindenum(x, context)
_sparsetensorcrdtransdirectionattr(x, context)
_sparsetensorsortkindattr(x, context)
_sparsetensorstoragespecifierkindattr(x, context)

Module Contents

mlir.dialects._sparse_tensor_enum_gen._ods_ir

class mlir.dialects._sparse_tensor_enum_gen.CrdTransDirectionKind

Bases: enum.IntEnum

sparse tensor coordinate translation direction

dim2lvl = 0

lvl2dim = 1

__str__()

Return str(self).

mlir.dialects._sparse_tensor_enum_gen._sparsetensorcrdtransdirectionenum(x, context)

class mlir.dialects._sparse_tensor_enum_gen.SparseTensorSortKind

Bases: enum.IntEnum

sparse tensor sort algorithm

HybridQuickSort = 0

InsertionSortStable = 1

QuickSort = 2

HeapSort = 3

__str__()

Return str(self).

mlir.dialects._sparse_tensor_enum_gen._sparsetensorsortkindenum(x, context)

class mlir.dialects._sparse_tensor_enum_gen.StorageSpecifierKind

Bases: enum.IntEnum

sparse tensor storage specifier kind

LvlSize = 0

PosMemSize = 1

CrdMemSize = 2

ValMemSize = 3

DimOffset = 4

DimStride = 5

__str__()

Return str(self).

mlir.dialects._sparse_tensor_enum_gen._sparsetensorstoragespecifierkindenum(x, context)

mlir.dialects._sparse_tensor_enum_gen._sparsetensorcrdtransdirectionattr(x, context)

mlir.dialects._sparse_tensor_enum_gen._sparsetensorsortkindattr(x, context)

mlir.dialects._sparse_tensor_enum_gen._sparsetensorstoragespecifierkindattr(x, context)