Lines Matching refs:ForOp

274 void ForOp::build(OpBuilder &builder, OperationState &result, Value lb,  in build()
292 ForOp::ensureTerminator(*bodyRegion, builder, result.location); in build()
301 LogicalResult ForOp::verify() { in verify()
318 LogicalResult ForOp::verifyRegions() { in verifyRegions()
352 Optional<Value> ForOp::getSingleInductionVar() { return getInductionVar(); } in getSingleInductionVar()
354 Optional<OpFoldResult> ForOp::getSingleLowerBound() { in getSingleLowerBound()
358 Optional<OpFoldResult> ForOp::getSingleStep() { in getSingleStep()
362 Optional<OpFoldResult> ForOp::getSingleUpperBound() { in getSingleUpperBound()
386 void ForOp::print(OpAsmPrinter &p) { in print()
401 ParseResult ForOp::parse(OpAsmParser &parser, OperationState &result) { in parse()
452 ForOp::ensureTerminator(*body, builder, result.location); in parse()
461 Region &ForOp::getLoopBody() { return getRegion(); } in getLoopBody()
463 ForOp mlir::scf::getForInductionVarOwner(Value val) { in getForInductionVarOwner()
466 return ForOp(); in getForInductionVarOwner()
469 return dyn_cast_or_null<ForOp>(containingOp); in getForInductionVarOwner()
476 OperandRange ForOp::getSuccessorEntryOperands(Optional<unsigned> index) { in getSuccessorEntryOperands()
489 void ForOp::getSuccessorRegions(Optional<unsigned> index, in getSuccessorRegions()
529 SmallVector<scf::ForOp, 4> loops; in buildLoopNest()
536 auto loop = builder.create<scf::ForOp>( in buildLoopNest()
607 struct ForOpIterArgsFolder : public OpRewritePattern<scf::ForOp> {
608 using OpRewritePattern<scf::ForOp>::OpRewritePattern;
610 LogicalResult matchAndRewrite(scf::ForOp forOp, in matchAndRewrite()
661 scf::ForOp newForOp = rewriter.create<scf::ForOp>( in matchAndRewrite()
721 struct SimplifyTrivialLoops : public OpRewritePattern<ForOp> {
722 using OpRewritePattern<ForOp>::OpRewritePattern;
724 LogicalResult matchAndRewrite(ForOp op, in matchAndRewrite()
781 static ForOp replaceTensorCastForOpIterArg(PatternRewriter &rewriter, in replaceTensorCastForOpIterArg()
789 ForOp forOp = cast<ForOp>(operand.getOwner()); in replaceTensorCastForOpIterArg()
804 scf::ForOp newForOp = rewriter.create<scf::ForOp>( in replaceTensorCastForOpIterArg()
873 struct ForOpTensorCastFolder : public OpRewritePattern<ForOp> {
874 using OpRewritePattern<ForOp>::OpRewritePattern;
876 LogicalResult matchAndRewrite(ForOp op, in matchAndRewrite()
964 struct LastTensorLoadCanonicalization : public OpRewritePattern<ForOp> {
965 using OpRewritePattern<ForOp>::OpRewritePattern;
967 LogicalResult matchAndRewrite(ForOp forOp, in matchAndRewrite()
1043 void ForOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns()
2275 ForOp::ensureTerminator(*body, builder, result.location); in parse()