MLIR
20.0.0git
|
#include "Dialect/SparseTensor/Transforms/Utils/SparseTensorDescriptor.h"
Public Member Functions | |
SparseTensorSpecifier (Value specifier) | |
operator Value () | |
Value | getSpecifierField (OpBuilder &builder, Location loc, StorageSpecifierKind kind, std::optional< Level > lvl) |
void | setSpecifierField (OpBuilder &builder, Location loc, Value v, StorageSpecifierKind kind, std::optional< Level > lvl) |
Static Public Member Functions | |
static Value | getInitValue (OpBuilder &builder, Location loc, SparseTensorType stt) |
Definition at line 24 of file SparseTensorDescriptor.h.
|
inlineexplicit |
Definition at line 26 of file SparseTensorDescriptor.h.
|
static |
Definition at line 81 of file SparseTensorDescriptor.cpp.
References mlir::OpBuilder::create(), mlir::get(), and mlir::sparse_tensor::SparseTensorType::getEncoding().
Referenced by createAllocFields().
Value SparseTensorSpecifier::getSpecifierField | ( | OpBuilder & | builder, |
Location | loc, | ||
StorageSpecifierKind | kind, | ||
std::optional< Level > | lvl | ||
) |
Definition at line 87 of file SparseTensorDescriptor.cpp.
References mlir::OpBuilder::create(), and optionalLevelAttr().
Referenced by mlir::sparse_tensor::SparseTensorDescriptorImpl< ValueArrayRef >::getSpecifierField().
|
inline |
Definition at line 33 of file SparseTensorDescriptor.h.
void SparseTensorSpecifier::setSpecifierField | ( | OpBuilder & | builder, |
Location | loc, | ||
Value | v, | ||
StorageSpecifierKind | kind, | ||
std::optional< Level > | lvl | ||
) |
Definition at line 94 of file SparseTensorDescriptor.cpp.
References mlir::OpBuilder::create(), mlir::Value::getType(), mlir::Type::isIndex(), and optionalLevelAttr().
Referenced by mlir::sparse_tensor::MutSparseTensorDescriptor::setSpecifierField().