SparseTensorStorageLayout.h File Reference

#include "mlir/Dialect/SparseTensor/IR/SparseTensor.h"
#include "mlir/Dialect/SparseTensor/IR/SparseTensorType.h"

Go to the source code of this file.

Classes
class	mlir::sparse_tensor::StorageLayout
	Provides methods to access fields of a sparse tensor with the given encoding. More...

Namespaces
	mlir
	Include the generated interface declarations.
	mlir::sparse_tensor

Typedefs
using	mlir::sparse_tensor::FieldIndex = unsigned
	The type of field indices. More...

Enumerations
enum class	mlir::sparse_tensor::SparseTensorFieldKind : uint32_t { mlir::sparse_tensor::StorageSpec = 0 , mlir::sparse_tensor::PosMemRef = static_cast<uint32_t>(StorageSpecifierKind::PosMemSize) , mlir::sparse_tensor::CrdMemRef = static_cast<uint32_t>(StorageSpecifierKind::CrdMemSize) , mlir::sparse_tensor::ValMemRef = static_cast<uint32_t>(StorageSpecifierKind::ValMemSize) }
	===-------------------------------------------------------------------—===// The sparse tensor storage scheme for a tensor is organized as a single compound type with the following fields. More...

Functions
StorageSpecifierKind	mlir::sparse_tensor::toSpecifierKind (SparseTensorFieldKind kind)
SparseTensorFieldKind	mlir::sparse_tensor::toFieldKind (StorageSpecifierKind kind)
unsigned	mlir::sparse_tensor::getNumFieldsFromEncoding (SparseTensorEncodingAttr enc)
unsigned	mlir::sparse_tensor::getNumDataFieldsFromEncoding (SparseTensorEncodingAttr enc)
void	mlir::sparse_tensor::foreachFieldInSparseTensor (SparseTensorEncodingAttr enc, llvm::function_ref< bool(FieldIndex, SparseTensorFieldKind, Level, LevelType)> callback)
void	mlir::sparse_tensor::foreachFieldAndTypeInSparseTensor (SparseTensorType, llvm::function_ref< bool(Type, FieldIndex, SparseTensorFieldKind, Level, LevelType)>)