MLIR
20.0.0git
|
#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) |