Lines Matching defs:padOp

486     auto padOp = packOp.getSource().getDefiningOp<tensor::PadOp>();
487     if (!padOp)
502     Value paddingVal = padOp.getConstantPaddingValue();
512     llvm::SmallBitVector paddedDims = padOp.getPaddedDims();
519     Location loc = padOp->getLoc();
521     rewriter.setInsertionPoint(padOp);
526         rewriter, loc, padOp.getSource(), mixedTiles, innerDimsPos,
529         loc, padOp.getSource(), empty, innerDimsPos, mixedTiles,
533     SmallVector<OpFoldResult> lowPad = padOp.getMixedLowPad();
534     SmallVector<OpFoldResult> highPad = padOp.getMixedHighPad();
547         padOp.getNofold());
551     if (!padOp->hasOneUse()) {
556       rewriter.replaceAllUsesExcept(padOp, unpackedPad, sourcePack);
1147   LogicalResult matchAndRewrite(tensor::PadOp padOp,
1150         padOp.getSource().getDefiningOp<tensor::UnPackOp>();
1154     if (!controlFn(&padOp.getSourceMutable()))
1157     Location loc = padOp.getLoc();
1159     llvm::SmallBitVector paddedDims = padOp.getPaddedDims();
1167     Value paddingVal = padOp.getConstantPaddingValue();
1173     SmallVector<OpFoldResult> lowPad = padOp.getMixedLowPad();
1174     SmallVector<OpFoldResult> highPad = padOp.getMixedHighPad();
1186         paddingVal, padOp.getNofold());
1188     // Inject the tensor.unpack right after the packed padOp.
1190         loc, padOp.getResultType().getShape(),
1191         padOp.getResultType().getElementType());
1196     rewriter.replaceOp(padOp, replacement);