Lines Matching defs:packOp

1473 static SmallVector<int64_t> getTiledPackShape(tensor::PackOp packOp,
1475   return applyPermutation(destShape, tensor::getPackInverseDestPerm(packOp));
1557 vectorizeAsTensorPackOp(RewriterBase &rewriter, tensor::PackOp packOp,
1561   rewriter.setInsertionPoint(packOp);
1563   Location loc = packOp.getLoc();
1564   auto padValue = packOp.getPaddingValue();
1567         loc, rewriter.getZeroAttr(packOp.getSourceType().getElementType()));
1571       cast<ReifyRankedShapedTypeOpInterface>(packOp.getOperation())
1581     ArrayRef<int64_t> resultTensorShape = packOp.getDestType().getShape();
1582     inputVectorSizes = resultTensorShape.take_front(packOp.getSourceRank());
1588   auto innerTiles = packOp.getStaticInnerTiles();
1589   auto innerDimsPos = packOp.getInnerDimsPos();
1590   auto outerDimsPerm = packOp.getOuterDimsPerm();
1597       rewriter, loc, packOp.getSource(), inputShape, padValue,
1603   auto tiledPackType = VectorType::get(getTiledPackShape(packOp, destShape),
1604                                        packOp.getDestType().getElementType());
1610       invertPermutationVector(tensor::getPackInverseDestPerm(packOp));
1945 vectorizePackOpPrecondition(tensor::PackOp packOp,
1947   auto padValue = packOp.getPaddingValue();
1950     LDBG("pad value is not constant: " << packOp << "\n");
1953   ArrayRef<int64_t> resultTensorShape = packOp.getDestType().getShape();
1956     if (!packOp.getDestType().hasStaticShape() ||
1957         !packOp.getSourceType().hasStaticShape())
1963           resultTensorShape.take_front(packOp.getSourceRank()),
1967   if (llvm::any_of(packOp.getInnerTiles(), [](OpFoldResult v) {
1970     LDBG("inner_tiles must be constant: " << packOp << "\n");
2141       .Case<tensor::PackOp>([&](auto packOp) {
2142         return vectorizePackOpPrecondition(packOp, inputVectorSizes);
2243           .Case<tensor::PackOp>([&](auto packOp) {
2244             return vectorizeAsTensorPackOp(rewriter, packOp, inputVectorSizes,