Lines Matching defs:enc

75 getSparseFieldShape(const SparseTensorEncodingAttr enc,
77   assert(enc);
80   SmallVector<int64_t> memrefShape(enc.getBatchLvlRank(), ShapedType::kDynamic);
85         enc.translateShape(*dimShape, CrdTransDirectionKind::dim2lvl);
87                        lvlShape.begin() + enc.getBatchLvlRank());
106   const auto lvlTypes = enc.getLvlTypes();
107   const Level lvlRank = enc.getLvlRank();
108   SmallVector<COOSegment> cooSegs = enc.getCOOSegments();
215     const Level cooStart = enc.getAoSCOOStart();
216     const Level lvlRank = enc.getLvlRank();
337 SparseTensorEncodingAttr::withDimToLvl(SparseTensorEncodingAttr enc) const {
338   return withDimToLvl(enc ? enc.getDimToLvl() : AffineMap());
1025   auto enc = SparseTensorEncodingAttr::get(
1028   return RankedTensorType::get(getDimShape(), getElementType(), enc);
1184 Dimension mlir::sparse_tensor::toDim(SparseTensorEncodingAttr enc, Level l) {
1185   if (enc) {
1186     assert(enc.isPermutation() && "Non permutation map not supported");
1187     if (const auto dimToLvl = enc.getDimToLvl())
1193 Level mlir::sparse_tensor::toLvl(SparseTensorEncodingAttr enc, Dimension d) {
1194   if (enc) {
1195     assert(enc.isPermutation() && "Non permutation map not supported");
1196     if (const auto lvlToDim = enc.getLvlToDim())
1207 getNormalizedEncodingForSpecifier(SparseTensorEncodingAttr enc) {
1209   for (auto lt : enc.getLvlTypes())
1213       enc.getContext(), lts,
1223       enc.getDimSlices());
1260   const auto enc = md.getType().getEncoding();
1261   const Level lvlRank = enc.getLvlRank();
1265     if (!enc.isSlice())
1276     if (mdKind == StorageSpecifierKind::PosMemSize && enc.isSingletonLvl(l))