MLIR  20.0.0git
Classes | Namespaces | Functions
SparseTensorDescriptor.h File Reference
#include "mlir/Dialect/SparseTensor/IR/SparseTensor.h"
#include "mlir/Dialect/SparseTensor/IR/SparseTensorStorageLayout.h"
#include "mlir/Dialect/SparseTensor/IR/SparseTensorType.h"
#include "mlir/Dialect/SparseTensor/Transforms/Passes.h"

Go to the source code of this file.

Classes

class  mlir::sparse_tensor::SparseTensorSpecifier
 
class  mlir::sparse_tensor::SparseTensorDescriptorImpl< ValueArrayRef >
 A helper class around an array of values that corresponds to a sparse tensor. More...
 
class  mlir::sparse_tensor::SparseTensorDescriptor
 Uses ValueRange for immutable descriptors. More...
 
class  mlir::sparse_tensor::MutSparseTensorDescriptor
 Using SmallVector for mutable descriptor allows users to reuse it as a tmp buffers to append value for some special cases, though users should be responsible to restore the buffer to legal states after their use. More...
 

Namespaces

 mlir
 Include the generated interface declarations.
 
 mlir::sparse_tensor
 

Functions

UnrealizedConversionCastOp mlir::sparse_tensor::getTuple (Value tensor)
 Returns the "tuple" value of the adapted tensor. More...
 
Value mlir::sparse_tensor::genTuple (OpBuilder &builder, Location loc, Type tp, ValueRange values)
 Packs the given values as a "tuple" value. More...
 
Value mlir::sparse_tensor::genTuple (OpBuilder &builder, Location loc, SparseTensorDescriptor desc)
 
SparseTensorDescriptor mlir::sparse_tensor::getDescriptorFromTensorTuple (Value tensor, RankedTensorType type)
 
MutSparseTensorDescriptor mlir::sparse_tensor::getMutDescriptorFromTensorTuple (Value tensor, SmallVectorImpl< Value > &fields, RankedTensorType type)