Lines Matching defs:stt

150     SparseTensorType stt,
154   assert(stt.hasEncoding());
157       getSparseFieldShape(stt.getEncoding(), stt.getDimShape());
159   const Type specType = StorageSpecifierType::get(stt.getEncoding());
161   const Type posMemType = MemRefType::get(memrefShape, stt.getPosType());
163   const Type crdMemType = MemRefType::get(memrefShape, stt.getCrdType());
165   const Type valMemType = MemRefType::get(memrefShape, stt.getElementType());
167   StorageLayout(stt).foreachField([specType, posMemType, crdMemType, valMemType,
1176     auto stt = tryGetSparseTensorType(v);
1177     return stt && !stt->isIdentity();
1283 static Type getFieldElemType(SparseTensorType stt, SparseTensorFieldKind kind) {
1286     return stt.getCrdType();
1288     return stt.getPosType();
1290     return stt.getElementType();
1298                                       SparseTensorType stt,
1301   if (requiresStaticShape && !stt.hasStaticDimShape())
1303   if (!stt.hasEncoding())
1307   Level cooStartLvl = stt.getAoSCOOStart();
1308   if (cooStartLvl < stt.getLvlRank()) {
1312     unsigned expCOORank = stt.getLvlRank() - cooStartLvl;
1319   StorageLayout layout(stt.getEncoding());
1325   layout.foreachField([&idx, &misMatch, stt, valTp,
1335       assert(fid == idx && stt.getLvlType(lvl) == lt);
1340     Type expElemTp = getFieldElemType(stt, fKind);
1494     auto stt = getSparseTensorType(getSource());
1495     if (static_cast<uint64_t>(lvl.value()) >= stt.getLvlRank())
1522   auto stt = getSparseTensorType(getSource());
1523   if (lvl >= stt.getLvlRank()) {
1535   SmallVector<Size> lvlShape = stt.getLvlShape();
1606   SparseTensorType stt = getSparseTensorType(adaptor.getTensor());
1610     elemTp = stt.getPosType();
1613     elemTp = stt.getCrdType();
1615       withStride = stt.getAoSCOOStart() <= adaptor.getLevel();
1617     elemTp = stt.getElementType();
1621   SmallVector<int64_t> bufShape = stt.getBatchLvlShape();
1625                                  stt.getContext(), ShapedType::kDynamic,
1633   auto stt = getSparseTensorType(getTensor());
1636   if (failed(isMatchingWidth(getResult(), stt.getPosWidth())))
1650   auto stt = getSparseTensorType(getTensor());
1653   if (failed(isMatchingWidth(getResult(), stt.getCrdWidth())))
1667   auto stt = getSparseTensorType(getTensor());
1668   if (stt.getAoSCOOStart() >= stt.getLvlRank())
1682   auto stt = getSparseTensorType(getTensor());
1684   if (stt.getElementType() != mtp.getElementType())
1925   const auto stt = getSparseTensorType(getTensor());
1926   if (stt.getLvlRank() != 1 + static_cast<Level>(getLvlCoords().size()))
1940   const auto stt = getSparseTensorType(tensor);
1941   const Dimension dimRank = stt.getDimRank();
1946   blockArgTypes.push_back(stt.getElementType());
2363   SparseTensorType stt = getSparseTensorType(adaptor.getTensor());
2364   ret.push_back(IterSpaceType::get(ctx, stt.getEncoding(), adaptor.getLoLvl(),
2394   auto stt = getSparseTensorType(getTensor());
2397   if (stt.getEncoding() != itTp.getEncoding())
2400   if (stt.getLvlRank() != itTp.getHiLvl())