Home
last modified time | relevance | path

Searched refs:dlt (Results 1 – 4 of 4) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/IR/
H A DSparseTensorDialect.cpp50 SmallVector<SparseTensorEncodingAttr::DimLevelType, 4> dlt; in parse() local
71 dlt.push_back(SparseTensorEncodingAttr::DimLevelType::Dense); in parse()
73 dlt.push_back(SparseTensorEncodingAttr::DimLevelType::Compressed); in parse()
75 dlt.push_back(SparseTensorEncodingAttr::DimLevelType::Singleton); in parse()
114 return parser.getChecked<SparseTensorEncodingAttr>(parser.getContext(), dlt, in parse()
/llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/
H A DCodegenUtils.h72 DimLevelType dimLevelTypeEncoding(SparseTensorEncodingAttr::DimLevelType dlt);
191 SparseTensorEncodingAttr::DimLevelType dlt) { in constantDimLevelTypeEncoding() argument
193 static_cast<uint8_t>(dimLevelTypeEncoding(dlt))); in constantDimLevelTypeEncoding()
H A DCodegenUtils.cpp145 SparseTensorEncodingAttr::DimLevelType dlt) { in dimLevelTypeEncoding() argument
146 switch (dlt) { in dimLevelTypeEncoding()
H A DSparseTensorConversion.cpp202 ArrayRef<SparseTensorEncodingAttr::DimLevelType> dlt = enc.getDimLevelType(); in newParams() local
203 unsigned sz = dlt.size(); in newParams()
207 attrs.push_back(constantDimLevelTypeEncoding(builder, loc, dlt[i])); in newParams()