Lines Matching refs:padTensorOp

1958   LogicalResult matchAndRewrite(PadOp padTensorOp,  in matchAndRewrite()
1960 if (!padTensorOp.hasZeroLowPad() || !padTensorOp.hasZeroHighPad()) in matchAndRewrite()
1962 if (padTensorOp.getNofold()) in matchAndRewrite()
1965 padTensorOp, padTensorOp.getResult().getType(), in matchAndRewrite()
1966 padTensorOp.getSource()); in matchAndRewrite()
1975 LogicalResult matchAndRewrite(PadOp padTensorOp, in matchAndRewrite()
1977 auto castOp = padTensorOp.getSource().getDefiningOp<tensor::CastOp>(); in matchAndRewrite()
1983 extractFromI64ArrayAttr(padTensorOp.getStaticLow()), in matchAndRewrite()
1984 extractFromI64ArrayAttr(padTensorOp.getStaticHigh()), in matchAndRewrite()
1985 padTensorOp.getResultType().getShape()); in matchAndRewrite()
1987 if (newResultType == padTensorOp.getResultType()) { in matchAndRewrite()
1988 rewriter.updateRootInPlace(padTensorOp, [&]() { in matchAndRewrite()
1989 padTensorOp.getSourceMutable().assign(castOp.getSource()); in matchAndRewrite()
1993 padTensorOp->getLoc(), newResultType, padTensorOp.getSource(), in matchAndRewrite()
1994 padTensorOp.getLow(), padTensorOp.getHigh(), in matchAndRewrite()
1995 padTensorOp.getStaticLow(), padTensorOp.getStaticHigh(), in matchAndRewrite()
1996 padTensorOp.getNofold()); in matchAndRewrite()
1998 padTensorOp.getRegion().cloneInto(&newOp.getRegion(), mapper); in matchAndRewrite()
2001 padTensorOp, padTensorOp.getResultType(), newOp); in matchAndRewrite()
2012 LogicalResult matchAndRewrite(PadOp padTensorOp, in matchAndRewrite()
2014 if (!padTensorOp.getResult().hasOneUse()) in matchAndRewrite()
2017 dyn_cast<tensor::CastOp>(*padTensorOp->getUsers().begin()); in matchAndRewrite()
2020 if (!tensor::preservesStaticInformation(padTensorOp.getResult().getType(), in matchAndRewrite()
2025 padTensorOp.getLoc(), tensorCastOp.getDest().getType(), in matchAndRewrite()
2026 padTensorOp.getSource(), padTensorOp.getLow(), padTensorOp.getHigh(), in matchAndRewrite()
2027 padTensorOp.getStaticLow(), padTensorOp.getStaticHigh(), in matchAndRewrite()
2028 padTensorOp.getNofold()); in matchAndRewrite()
2029 replacementOp.getRegion().takeBody(padTensorOp.getRegion()); in matchAndRewrite()
2031 rewriter.replaceOp(padTensorOp, replacementOp.getResult()); in matchAndRewrite()