Lines Matching defs:loopState

416                                   ValueRange /*loopState*/) {
443                                     ValueRange /*loopState*/) {
504                                    Value iv, ValueRange loopState) {
515     auto source = loopState.empty() ? xferOp.getSource() : loopState[0];
530                                     ValueRange loopState) {
531     return isTensorOp(xferOp) ? loopState[0] : Value();
947     auto loopState = Strategy<OpTy>::initialLoopState(xferOp);
951         lb, ub, step, loopState ? ValueRange(loopState) : ValueRange(),
952         [&](OpBuilder &b, Location loc, Value iv, ValueRange loopState) {
953           Type stateType = loopState.empty() ? Type() : loopState[0].getType();
962                     b, this->options, xferOp, castedDataBuffer, iv, loopState);
983                 return loopState.empty() ? Value() : newXfer->getResult(0);
988                     b, xferOp, castedDataBuffer, iv, loopState);
991           maybeYieldValue(b, loc, !loopState.empty(), result);
1553                                   ValueRange loopState) {
1556     auto vec = loopState[0];
1586                                   ValueRange /*loopState*/) {
1668     auto loopState = Strategy1d<OpTy>::initialLoopState(rewriter, xferOp);
1672         xferOp, lb, ub, step, loopState ? ValueRange(loopState) : ValueRange(),
1673         [&](OpBuilder &b, Location loc, Value iv, ValueRange loopState) {
1674           Strategy1d<OpTy>::generateForLoopBody(b, loc, xferOp, iv, loopState);