MLIR  17.0.0git
Namespaces | Macros | Functions
SparseTensor.h File Reference
#include "mlir/Dialect/SparseTensor/IR/Enums.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/InferTypeOpInterface.h"
#include "mlir/Interfaces/SideEffectInterfaces.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"
+ Include dependency graph for SparseTensor.h:
+ This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces

 mlir
 Include the generated interface declarations.
 
 mlir::sparse_tensor
 

Macros

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

Functions

template<typename T >
RankedTensorType mlir::sparse_tensor::getRankedTensorType (T t)
 Convenience method to abbreviate casting getType(). More...
 
template<typename T >
MemRefType mlir::sparse_tensor::getMemRefType (T t)
 Convenience method to abbreviate casting getType(). More...
 
SparseTensorEncodingAttr mlir::sparse_tensor::getSparseTensorEncoding (Type type)
 Convenience method to get a sparse encoding attribute from a type. More...
 
bool mlir::sparse_tensor::isUniqueCOOType (RankedTensorType tp)
 Returns true iff the given type is a type for a COO tensor with the last dimension level type being unique. More...
 
unsigned mlir::sparse_tensor::getCOOStart (SparseTensorEncodingAttr enc)
 Returns the starting dimension for a trailing COO region that spans across at least two dimensions. More...
 
DimLevelType mlir::sparse_tensor::getDimLevelType (SparseTensorEncodingAttr enc, uint64_t d)
 
DimLevelType mlir::sparse_tensor::getDimLevelType (RankedTensorType type, uint64_t d)
 
bool mlir::sparse_tensor::isDenseDim (RankedTensorType type, uint64_t d)
 Convenience function to test for dense dimension (0 <= d < rank). More...
 
bool mlir::sparse_tensor::isCompressedDim (RankedTensorType type, uint64_t d)
 Convenience function to test for compressed dimension (0 <= d < rank). More...
 
bool mlir::sparse_tensor::isSingletonDim (RankedTensorType type, uint64_t d)
 Convenience function to test for singleton dimension (0 <= d < rank). More...
 
bool mlir::sparse_tensor::isDenseDim (SparseTensorEncodingAttr enc, uint64_t d)
 Convenience function to test for dense dimension (0 <= d < rank). More...
 
bool mlir::sparse_tensor::isCompressedDim (SparseTensorEncodingAttr enc, uint64_t d)
 Convenience function to test for compressed dimension (0 <= d < rank). More...
 
bool mlir::sparse_tensor::isSingletonDim (SparseTensorEncodingAttr enc, uint64_t d)
 Convenience function to test for singleton dimension (0 <= d < rank). More...
 
bool mlir::sparse_tensor::isOrderedDim (RankedTensorType type, uint64_t d)
 Convenience function to test for ordered property in the given dimension (0 <= d < rank). More...
 
bool mlir::sparse_tensor::isUniqueDim (RankedTensorType type, uint64_t d)
 Convenience function to test for unique property in the given dimension (0 <= d < rank). More...
 
uint64_t mlir::sparse_tensor::toOrigDim (SparseTensorEncodingAttr enc, uint64_t d)
 
uint64_t mlir::sparse_tensor::toStoredDim (SparseTensorEncodingAttr enc, uint64_t d)
 
uint64_t mlir::sparse_tensor::toOrigDim (RankedTensorType type, uint64_t d)
 Convenience method to translate the given stored dimension to the original dimension (0 <= d < rank). More...
 
uint64_t mlir::sparse_tensor::toStoredDim (RankedTensorType type, uint64_t d)
 Convenience method to translate the given original dimension to the stored dimension (0 <= d < rank). More...
 

Macro Definition Documentation

◆ GET_ATTRDEF_CLASSES [1/2]

#define GET_ATTRDEF_CLASSES

Definition at line 27 of file SparseTensor.h.

◆ GET_ATTRDEF_CLASSES [2/2]

#define GET_ATTRDEF_CLASSES

Definition at line 27 of file SparseTensor.h.

◆ GET_OP_CLASSES

#define GET_OP_CLASSES

Definition at line 33 of file SparseTensor.h.

◆ GET_TYPEDEF_CLASSES

#define GET_TYPEDEF_CLASSES

Definition at line 30 of file SparseTensor.h.