Lines Matching defs:yield
84 /// Default callback for IfOp builders. Inserts a yield without arguments.
97 if (auto yield = dyn_cast_or_null<TerminatorTy>(terminatorOperation))
98 return yield;
171 // scf.yield %x : i64
207 // scf.yield %z : i64
558 // Generate the new yield values and append them to the scf.yield operation.
568 "expected as many new yield values as new iter operands");
735 // For all loops but the innermost, yield the results of the nested loop.
742 // and yield its results.
818 // 5. Inject an outgoing cast op at the end of the block and yield it instead.
878 forOp.getYieldedValues() // iter yield
1028 // defined outside of the loop, remove it and replace it with yield values.
1045 /// scf.yield %2 : tensor<?x?xf32>
1058 /// scf.yield %4 : tensor<32x1024xf32>
2139 // Print yield explicitly if the op defines values.
2240 // Replace the yield op by one that returns only the used values.
2436 /// yield something(), true
2438 /// yield something2(), false
2444 /// yield something()
2446 /// yield something2()
2450 /// Additionally if both branches yield the same value, replace all uses
2454 /// yield something(), %arg1
2456 /// yield something2(), %arg1
2462 /// yield something()
2464 /// yield something2()
2706 /// scf.yield
2708 /// scf.yield
2715 /// scf.yield
2727 // If there is an else block, it can only yield
2761 // If the correctness test passes, we will yield
2772 // the else region contains just the yield, its yielded value is
3460 "expects the 'after' region to terminate with 'scf.yield'");
3517 /// 1. i-th yield operand is equal to the i-th while operand.
3518 /// 2. i-th yield operand is k-th after block argument which is (k+1)-th
3536 /// scf.yield %arg0_after_2, %b, %arg1_after, ..., %argN
3549 /// scf.yield %arg1_after, ..., %argN
3553 /// We iterate over each yield operand.
3554 /// 1. 0-th yield operand %arg0_after_2 is 4-th condition operand
3556 /// remove 0-th before block argument and yield operand, and replace
3559 /// 2. 1-th yield operand %b is equal to the 1-th iter arg's initial
3581 // If i-th yield operand is equal to the i-th operand of the scf.while,
3587 // If the i-th yield operand is k-th after block argument, then we check
3613 // If i-th yield operand is equal to the i-th operand of the scf.while,
3619 // If the i-th yield operand is k-th after block argument, then we check
3694 /// scf.yield %arg0_after, %arg2_after, ..., %argN_after
3706 /// scf.yield %arg0, %b, ..., %argN
3816 /// scf.yield
3829 /// scf.yield
3963 /// Remove unused init/yield args.
3974 YieldOp yield = op.getYieldOp();
3986 op.getBeforeArguments(), yield.getOperands(), op.getInits())) {
4009 rewriter.setInsertionPoint(yield);
4010 rewriter.replaceOpWithNewOp<YieldOp>(yield, newYields);
4223 auto yield = dyn_cast<YieldOp>(region.front().back());
4224 if (!yield)
4225 return emitOpError("expected region to end with scf.yield, but got ")
4228 if (yield.getNumOperands() != getNumResults()) {
4231 << yield.getNumOperands())
4232 .attachNote(yield.getLoc())
4233 << "see yield operation here";
4237 yield.getOperandTypes())) {
4242 .attachNote(yield.getLoc())