Lines Matching refs:loopState
482 Value iv, ValueRange loopState) { in rewriteOp()
493 auto source = loopState.empty() ? xferOp.getSource() : loopState[0]; in rewriteOp()
508 ValueRange loopState) { in handleOutOfBoundsDim()
509 return isTensorOp(xferOp) ? loopState[0] : Value(); in handleOutOfBoundsDim()
725 auto loopState = Strategy<OpTy>::initialLoopState(xferOp); in matchAndRewrite() local
729 lb, ub, step, loopState ? ValueRange(loopState) : ValueRange(), in matchAndRewrite()
730 [&](OpBuilder &b, Location loc, Value iv, ValueRange loopState) { in matchAndRewrite() argument
731 Type stateType = loopState.empty() ? Type() : loopState[0].getType(); in matchAndRewrite()
740 b, this->options, xferOp, castedDataBuffer, iv, loopState); in matchAndRewrite()
766 return loopState.empty() ? Value() : newXfer->getResult(0); in matchAndRewrite()
771 b, xferOp, castedDataBuffer, iv, loopState); in matchAndRewrite()
774 maybeYieldValue(b, loc, !loopState.empty(), result); in matchAndRewrite()
1126 ValueRange loopState) { in generateForLoopBody()
1129 auto vec = loopState[0]; in generateForLoopBody()
1244 auto loopState = Strategy1d<OpTy>::initialLoopState(rewriter, xferOp); in matchAndRewrite() local
1248 xferOp, lb, ub, step, loopState ? ValueRange(loopState) : ValueRange(), in matchAndRewrite()
1249 [&](OpBuilder &b, Location loc, Value iv, ValueRange loopState) { in matchAndRewrite() argument
1250 Strategy1d<OpTy>::generateForLoopBody(b, loc, xferOp, iv, loopState); in matchAndRewrite()