|
MLIR
22.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 79 of file SparseTensorDescriptor.cpp.
References 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 85 of file SparseTensorDescriptor.cpp.
References kind, 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 93 of file SparseTensorDescriptor.cpp.
References mlir::Value::getType(), mlir::Type::isIndex(), kind, and optionalLevelAttr().
Referenced by mlir::sparse_tensor::MutSparseTensorDescriptor::setSpecifierField().