Lines Matching defs:innerDimsPos
77 ArrayRef<int64_t> innerDimsPos = packOrUnPackOp.getInnerDimsPos();
79 llvm::zip_equal(llvm::seq<unsigned>(0, innerDimsPos.size()),
80 innerDimsPos, packOrUnPackOp.getMixedTiles())) {
250 SmallVector<int64_t> innerDimsPos;
257 innerDimsPos.push_back(index);
290 if (innerDimsPos.empty() && outerDimsPerm.empty())
294 b, loc, opOperand->get(), innerTileSizes, innerDimsPos, outerDimsPerm);
296 loc, opOperand->get(), empty, innerDimsPos, innerTileSizes,
509 ArrayRef<int64_t> innerDimsPos = packOp.getInnerDimsPos();
514 for (int64_t dim : innerDimsPos)
526 rewriter, loc, padOp.getSource(), mixedTiles, innerDimsPos,
529 loc, padOp.getSource(), empty, innerDimsPos, mixedTiles,
541 size_t pointLoopsSize = innerDimsPos.size();
553 rewriter, loc, newPadOp, mixedTiles, innerDimsPos, outerDimsPerm);
555 loc, newPadOp, unpackEmpty, innerDimsPos, mixedTiles, outerDimsPerm);
655 ArrayRef<int64_t> innerDimsPos = packOp.getInnerDimsPos();
669 projectToInnerMostNonUnitDimsPos(innerDimsPos, reassocIndices, srcShape);
699 for (size_t i = 0; i < innerDimsPos.size(); ++i) {
901 ArrayRef<int64_t> innerDimsPos = unPackOp.getInnerDimsPos();
918 projectToInnerMostNonUnitDimsPos(innerDimsPos, reassocIndices, dstShape);
941 for (size_t i = 0; i < innerDimsPos.size(); ++i) {
1105 auto innerDimsPos = destPack.getInnerDimsPos();
1112 destPack.getSource(), innerDimsPos,
1160 ArrayRef<int64_t> innerDimsPos = unpackOp.getInnerDimsPos();
1162 for (int64_t dim : innerDimsPos)
1180 size_t pointLoopsSize = innerDimsPos.size();
1194 loc, newPadOp.getResult(), outputUnPack, innerDimsPos,