Lines Matching defs:unpackOp

1632 vectorizeAsTensorUnpackOp(RewriterBase &rewriter, tensor::UnPackOp unpackOp,
1637   rewriter.setInsertionPoint(unpackOp);
1639   RankedTensorType unpackTensorType = unpackOp.getSourceType();
1641   ArrayRef<int64_t> innerDimPos = unpackOp.getInnerDimsPos();
1642   ArrayRef<int64_t> innerTiles = unpackOp.getStaticInnerTiles();
1645   ArrayRef<int64_t> outerDimsPerm = unpackOp.getOuterDimsPerm();
1647   auto destSize = unpackOp.getDestRank();
1705       cast<ReifyRankedShapedTypeOpInterface>(unpackOp.getOperation())
1708     LDBG("Unable to reify result shapes of " << unpackOp);
1711   Location loc = unpackOp->getLoc();
1714       loc, rewriter.getZeroAttr(unpackOp.getSourceType().getElementType()));
1719       rewriter, loc, unpackOp.getSource(), readVectorSizes, padValue,
1724       tensor::getUnPackInverseSrcPerm(unpackOp, packMetadata);
1746       unpackOp.getDestType().hasStaticShape()
1857 vectorizeUnPackOpPrecondition(tensor::UnPackOp unpackOp,
1860   if (llvm::any_of(unpackOp.getInnerTiles(), [](OpFoldResult res) {
1863     LDBG("Inner-tiles must be constant: " << unpackOp << "\n");
1866   ArrayRef<int64_t> resultShape = unpackOp.getDestType().getShape();
1868                           unpackOp.getDestType().hasStaticShape() &&
1869                           unpackOp.getSourceType().hasStaticShape();
2144       .Case<tensor::UnPackOp>([&](auto unpackOp) {
2145         return vectorizeUnPackOpPrecondition(unpackOp, inputVectorSizes);
2247           .Case<tensor::UnPackOp>([&](auto unpackOp) {
2248             return vectorizeAsTensorUnpackOp(rewriter, unpackOp,