SparseTensor.h File Reference

#include "mlir/Bytecode/BytecodeOpInterface.h"
#include "mlir/Dialect/SparseTensor/IR/Enums.h"
#include "mlir/Dialect/SparseTensor/IR/SparseTensorInterfaces.h"
#include "mlir/IR/BuiltinTypes.h"
#include "mlir/IR/Dialect.h"
#include "mlir/IR/OpDefinition.h"
#include "mlir/IR/OpImplementation.h"
#include "mlir/IR/TensorEncoding.h"
#include "mlir/Interfaces/ControlFlowInterfaces.h"
#include "mlir/Interfaces/InferTypeOpInterface.h"
#include "mlir/Interfaces/LoopLikeInterface.h"
#include "mlir/Interfaces/SideEffectInterfaces.h"
#include "llvm/ADT/bit.h"
#include "mlir/Dialect/SparseTensor/IR/SparseTensorAttrEnums.h.inc"
#include "mlir/Dialect/SparseTensor/IR/SparseTensorAttrDefs.h.inc"
#include "mlir/Dialect/SparseTensor/IR/SparseTensorTypes.h.inc"
#include "mlir/Dialect/SparseTensor/IR/SparseTensorOps.h.inc"
#include "mlir/Dialect/SparseTensor/IR/SparseTensorOpsDialect.h.inc"

Go to the source code of this file.

Classes
struct	mlir::sparse_tensor::COOSegment
	A simple structure that encodes a range of levels in the sparse tensors that forms a COO segment. More...
class	mlir::sparse_tensor::I64BitSet
	A simple wrapper to encode a bitset of (at most 64) levels, currently used by sparse_tensor.iterate operation for the set of levels on which the coordinates should be loaded. More...

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

Macros
#define	GET_ATTRDEF_CLASSES
#define	GET_ATTRDEF_CLASSES
#define	GET_TYPEDEF_CLASSES
#define	GET_OP_CLASSES

Typedefs
using	mlir::sparse_tensor::Dimension = uint64_t
	The type of dimension identifiers and dimension-ranks.
using	mlir::sparse_tensor::Level = uint64_t
	The type of level identifiers and level-ranks.
using	mlir::sparse_tensor::Size = int64_t
	The type for individual components of a compile-time shape, including the value ShapedType::kDynamic (for shapes).

Functions
template<typename T>
RankedTensorType	mlir::sparse_tensor::getRankedTensorType (T &&t)
	Convenience method to abbreviate casting getType().
template<typename T>
MemRefType	mlir::sparse_tensor::getMemRefType (T &&t)
	Convenience method to abbreviate casting getType().
SparseTensorEncodingAttr	mlir::sparse_tensor::getSparseTensorEncoding (Type type)
	Convenience method to get a sparse encoding attribute from a type.
bool	mlir::sparse_tensor::hasAnySparseType (TypeRange types)
	Returns true iff the type range has any sparse tensor type.
bool	mlir::sparse_tensor::hasAnySparseOperand (Operation *op)
	Returns true iff MLIR operand has any sparse operand.
bool	mlir::sparse_tensor::hasAnySparseResult (Operation *op)
	Returns true iff MLIR operand has any sparse result.
bool	mlir::sparse_tensor::hasAnySparseOperandOrResult (Operation *op)
	Returns true iff MLIR operand has any sparse operand or result.
bool	mlir::sparse_tensor::hasAnyNonIdentityOperandsOrResults (Operation *op)
	Returns true iff MLIR operation has any sparse tensor with non-identity dim2lvl maps.
AffineMap	mlir::sparse_tensor::inferLvlToDim (AffineMap dimToLvl, MLIRContext *context)
	Given the dimToLvl map, infers the lvlToDim map, or returns empty Affine map when inference fails.
AffineMap	mlir::sparse_tensor::inverseBlockSparsity (AffineMap dimToLvl, MLIRContext *context)
	Returns the lvlToDim map for the given dimToLvl map specific to the block sparse cases.
SmallVector< unsigned >	mlir::sparse_tensor::getBlockSize (AffineMap dimToLvl)
	Given the dimToLvl map, returns the block sizes in a vector.
bool	mlir::sparse_tensor::isBlockSparsity (AffineMap dimToLvl)
	Given the dimToLvl map, returns if it's block sparsity.
Dimension	mlir::sparse_tensor::toDim (SparseTensorEncodingAttr enc, Level l)
	Convenience method to translate the given level to the corresponding dimension.
Level	mlir::sparse_tensor::toLvl (SparseTensorEncodingAttr enc, Dimension d)
	Convenience method to translate the given dimension to the corresponding level.

Macro Definition Documentation

GET_ATTRDEF_CLASSES [1/2]

#define GET_ATTRDEF_CLASSES

Definition at line 137 of file SparseTensor.h.

GET_ATTRDEF_CLASSES [2/2]

#define GET_ATTRDEF_CLASSES

Definition at line 137 of file SparseTensor.h.

GET_OP_CLASSES

#define GET_OP_CLASSES

Definition at line 146 of file SparseTensor.h.

GET_TYPEDEF_CLASSES

#define GET_TYPEDEF_CLASSES

Definition at line 143 of file SparseTensor.h.