150     if (!isa<AffineParallelOp, AffineForOp, AffineIfOp>(destOp))  in isLegalToInline()
198            isa<AffineForOp, AffineParallelOp, AffineIfOp>(parentOp);  in isLegalToInline()
272                       isa<AffineForOp, AffineParallelOp>(parentOp));  in isValidDim()
294     return isa<AffineForOp, AffineParallelOp>(parentOp);  in isValidDim()
1430 void AffineForOp::build(OpBuilder &builder, OperationState &result,  in build()
1481 void AffineForOp::build(OpBuilder &builder, OperationState &result, int64_t lb,  in build()
1490 LogicalResult AffineForOp::verifyRegions() {  in verifyRegions()
1536   auto boundAttrStrName = isLower ? AffineForOp::getLowerBoundAttrStrName()  in parseBound()
1537                                   : AffineForOp::getUpperBoundAttrStrName();  in parseBound()
1619 ParseResult AffineForOp::parse(OpAsmParser &parser, OperationState &result) {  in parse()
1636         AffineForOp::getStepAttrStrName(),  in parse()
1642                               AffineForOp::getStepAttrStrName().data(),  in parse()
1684   AffineForOp::ensureTerminator(*body, builder, result.location);  in parse()
1731 unsigned AffineForOp::getNumIterOperands() {  in getNumIterOperands()
1738 void AffineForOp::print(OpAsmPrinter &p) {  in print()
1773 static LogicalResult foldLoopBounds(AffineForOp forOp) {  in foldLoopBounds()
1819 static LogicalResult canonicalizeLoopBounds(AffineForOp forOp) {  in canonicalizeLoopBounds()
1849 static Optional<uint64_t> getTrivialConstantTripCount(AffineForOp forOp) {  in getTrivialConstantTripCount()
1860 struct AffineForEmptyLoopFolder : public OpRewritePattern<AffineForOp> {
1861   using OpRewritePattern<AffineForOp>::OpRewritePattern;
1863   LogicalResult matchAndRewrite(AffineForOp forOp,  in matchAndRewrite()
1913 void AffineForOp::getCanonicalizationPatterns(RewritePatternSet &results,  in getCanonicalizationPatterns()
1922 OperandRange AffineForOp::getSuccessorEntryOperands(Optional<unsigned> index) {  in getSuccessorEntryOperands()
1935 void AffineForOp::getSuccessorRegions(  in getSuccessorRegions()
1969 static bool hasTrivialZeroTripCount(AffineForOp op) {  in hasTrivialZeroTripCount()
1974 LogicalResult AffineForOp::fold(ArrayRef<Attribute> operands,  in fold()
1987 AffineBound AffineForOp::getLowerBound() {  in getLowerBound()
1989   return AffineBound(AffineForOp(*this), 0, lbMap.getNumInputs(), lbMap);  in getLowerBound()
1992 AffineBound AffineForOp::getUpperBound() {  in getUpperBound()
1995   return AffineBound(AffineForOp(*this), lbMap.getNumInputs(),  in getUpperBound()
1999 void AffineForOp::setLowerBound(ValueRange lbOperands, AffineMap map) {  in setLowerBound()
2014 void AffineForOp::setUpperBound(ValueRange ubOperands, AffineMap map) {  in setUpperBound()
2027 void AffineForOp::setLowerBoundMap(AffineMap map) {  in setLowerBoundMap()
2036 void AffineForOp::setUpperBoundMap(AffineMap map) {  in setUpperBoundMap()
2045 bool AffineForOp::hasConstantLowerBound() {  in hasConstantLowerBound()
2049 bool AffineForOp::hasConstantUpperBound() {  in hasConstantUpperBound()
2053 int64_t AffineForOp::getConstantLowerBound() {  in getConstantLowerBound()
2057 int64_t AffineForOp::getConstantUpperBound() {  in getConstantUpperBound()
2061 void AffineForOp::setConstantLowerBound(int64_t value) {  in setConstantLowerBound()
2065 void AffineForOp::setConstantUpperBound(int64_t value) {  in setConstantUpperBound()
2069 AffineForOp::operand_range AffineForOp::getLowerBoundOperands() {  in getLowerBoundOperands()
2073 AffineForOp::operand_range AffineForOp::getUpperBoundOperands() {  in getUpperBoundOperands()
2079 AffineForOp::operand_range AffineForOp::getControlOperands() {  in getControlOperands()
2084 bool AffineForOp::matchingBoundOperandList() {  in matchingBoundOperandList()
2100 Region &AffineForOp::getLoopBody() { return getRegion(); }  in getLoopBody()
2102 Optional<Value> AffineForOp::getSingleInductionVar() {  in getSingleInductionVar()
2106 Optional<OpFoldResult> AffineForOp::getSingleLowerBound() {  in getSingleLowerBound()
2113 Optional<OpFoldResult> AffineForOp::getSingleStep() {  in getSingleStep()
2118 Optional<OpFoldResult> AffineForOp::getSingleUpperBound() {  in getSingleUpperBound()
2128   return getForInductionVarOwner(val) != AffineForOp();  in isForInductionVar()
2133 AffineForOp mlir::getForInductionVarOwner(Value val) {  in getForInductionVarOwner()
2136     return AffineForOp();  in getForInductionVarOwner()
2138   if (auto forOp = dyn_cast<AffineForOp>(containingInst))  in getForInductionVarOwner()
2140     return forOp.getInductionVar() == val ? forOp : AffineForOp();  in getForInductionVarOwner()
2141   return AffineForOp();  in getForInductionVarOwner()
2146 void mlir::extractForInductionVars(ArrayRef<AffineForOp> forInsts,  in extractForInductionVars()
2196 static AffineForOp
2199                              AffineForOp::BodyBuilderFn bodyBuilderFn) {  in buildAffineLoopFromConstants()
2200   return builder.create<AffineForOp>(loc, lb, ub, step, /*iterArgs=*/llvm::None,  in buildAffineLoopFromConstants()
2205 static AffineForOp
2208                           AffineForOp::BodyBuilderFn bodyBuilderFn) {  in buildAffineLoopFromValues()
2214   return builder.create<AffineForOp>(loc, lb, builder.getDimIdentityMap(), ub,  in buildAffineLoopFromValues()
2235 AffineForOp mlir::replaceForOpWithNewYields(OpBuilder &b, AffineForOp loop,  in replaceForOpWithNewYields()
2252   AffineForOp newLoop =  in replaceForOpWithNewYields()
2253       b.create<AffineForOp>(loop.getLoc(), lbOperands, lbMap, ubOperands, ubMap,  in replaceForOpWithNewYields()
3834   if (!isa<AffineParallelOp, AffineIfOp, AffineForOp>(parentOp))  in verify()