Lines Matching defs:dimRank
611 const Dimension dimRank = dlm.getDimRank();
612 for (Dimension dim = 0; dim < dimRank; dim++)
623 for (Dimension dim = 0; dim < dimRank; dim++)
870 // We save `dimRank` here because we'll also need it to verify `dimSlices`.
871 const Dimension dimRank = dimToLvl ? dimToLvl.getNumDims() : lvlRank;
883 if (dimRank > lvlRank)
884 return emitError() << "unexpected dimToLvl mapping from " << dimRank
888 if (dimSlices.size() != dimRank)
891 << dimSlices.size() << " != " << dimRank;
894 if (dimRank != lvlRank)
897 << dimRank << " != " << lvlRank;
914 const Dimension dimRank = dimShape.size();
915 if (dimRank == 0)
917 if (getDimRank() != dimRank)
920 << getDimRank() << " != " << dimRank;
1389 for (Dimension d = 0, dimRank = tp1.getRank(); d < dimRank; d++)
1855 const Dimension dimRank = dstTp.getDimRank();
1860 if (concatDim >= dimRank)
1863 concatDim, dimRank));
1871 if (srcDimRank != dimRank)
1875 i, srcDimRank, dimRank));
1878 for (Dimension d = 0; d < dimRank; d++) {
1941 const Dimension dimRank = stt.getDimRank();
1943 // Starts with `dimRank`-many coordinates.
1944 SmallVector<Type> blockArgTypes(dimRank, builder.getIndexType());
1957 bodyBlock->getArguments().slice(0, dimRank),
1958 bodyBlock->getArguments()[dimRank],
1959 bodyBlock->getArguments().drop_front(dimRank + 1));
1964 const Dimension dimRank = t.getDimRank();
1970 if (dimRank + 1 + getInitArgs().size() != args.size())
1986 for (Dimension d = 0; d < dimRank; d++)
1992 const auto valueTp = args[dimRank].getType();