Lines Matching defs:padOp

1014   LogicalResult matchAndRewrite(tensor::PadOp padOp,
1017         padOp.getSource().getDefiningOp<tensor::CollapseShapeOp>();
1023     if (!controlFoldingReshapes(&padOp.getSourceMutable())) {
1024       return rewriter.notifyMatchFailure(padOp,
1028     ArrayRef<int64_t> low = padOp.getStaticLow();
1029     ArrayRef<int64_t> high = padOp.getStaticHigh();
1040     RankedTensorType paddedType = padOp.getResultType();
1047         newLow.push_back(padOp.getMixedLowPad()[idx]);
1048         newHigh.push_back(padOp.getMixedHighPad()[idx]);
1052     Location loc = padOp->getLoc();
1056         padOp.getConstantPaddingValue(), padOp.getNofold());
1059         padOp, padOp.getResultType(), newPadOp.getResult(), reassociations);
1823   LogicalResult matchAndRewrite(tensor::PadOp padOp,
1826         padOp.getSource().getDefiningOp<tensor::ExpandShapeOp>();
1832     if (!controlFoldingReshapes(&padOp.getSourceMutable())) {
1833       return rewriter.notifyMatchFailure(padOp,
1837     ArrayRef<int64_t> low = padOp.getStaticLow();
1838     ArrayRef<int64_t> high = padOp.getStaticHigh();
1854     RankedTensorType paddedType = padOp.getResultType();
1864       OpFoldResult l = padOp.getMixedLowPad()[reInd[0]];
1865       OpFoldResult h = padOp.getMixedHighPad()[reInd[0]];
1880         padOp.getConstantPaddingValue(), padOp.getNofold());
1883         padOp, padOp.getResultType(), newPadOp.getResult(), reassociations,