Lines Matching defs:forOp

45   static void populateIterArgBounds(scf::ForOp forOp, Value value,
51       iterArgIdx = iterArg.getArgNumber() - forOp.getNumInductionVars();
56     Value yieldedValue = cast<scf::YieldOp>(forOp.getBody()->getTerminator())
58     Value iterArg = forOp.getRegionIterArg(iterArgIdx);
59     Value initArg = forOp.getInitArgs()[iterArgIdx];
77     // `value` is result of `forOp`, we can prove that:
80     AffineExpr lbExpr = cstr.getExpr(forOp.getLowerBound());
81     AffineExpr ubExpr = cstr.getExpr(forOp.getUpperBound());
82     AffineExpr stepExpr = cstr.getExpr(forOp.getStep());
93     auto forOp = cast<ForOp>(op);
95     if (value == forOp.getInductionVar()) {
97       cstr.bound(value) >= forOp.getLowerBound();
98       cstr.bound(value) < forOp.getUpperBound();
103     populateIterArgBounds(forOp, value, std::nullopt, cstr);
108     auto forOp = cast<ForOp>(op);
110     populateIterArgBounds(forOp, value, dim, cstr);