MLIR  20.0.0git
Classes | Namespaces | Typedefs | Enumerations | Functions
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)>)