Lines Matching refs:forOp

243   for (scf::ForOp forOp : llvm::reverse(reverseEnclosingLoops))  in HoistingAnalysis()  local
244 if (backwardSlice.contains(forOp)) in HoistingAnalysis()
245 packingLoops.push_back(forOp); in HoistingAnalysis()
306 if (auto forOp = dyn_cast<scf::ForOp>(op)) { in dropNonIndexDependencies() local
307 if (!hasIndexResult(op) && indexEdges.contains(forOp.getInductionVar())) { in dropNonIndexDependencies()
351 for (auto forOp : packingLoops) { in getPackedTensorSizes() local
355 getUpperBoundForIndex(forOp.getUpperBound(), boundMap, boundOperands); in getPackedTensorSizes()
365 (ub - lb).ceilDiv(step), ValueRange{forOp.getLowerBound(), ubVal, in getPackedTensorSizes()
366 cast<scf::ForOp>(forOp).getStep()}); in getPackedTensorSizes()
382 scf::ForOp forOp) { in buildLoopIterationCount() argument
383 MLIRContext *ctx = forOp->getContext(); in buildLoopIterationCount()
387 if (!isDefinedOutsideOrConstant(outer, forOp.getLowerBound()) || in buildLoopIterationCount()
388 !isDefinedOutsideOrConstant(outer, forOp.getStep())) in buildLoopIterationCount()
390 Value ivVal = forOp.getInductionVar(), lbVal = forOp.getLowerBound(), in buildLoopIterationCount()
391 stepVal = forOp.getStep(); in buildLoopIterationCount()
392 auto loc = forOp->getLoc(); in buildLoopIterationCount()
459 auto forOp = dyn_cast<scf::ForOp>(op); in hoistPaddingOnTensors() local
460 if (!forOp) { in hoistPaddingOnTensors()
466 loc, bvm.lookupOrDefault(forOp.getLowerBound()), in hoistPaddingOnTensors()
467 bvm.lookupOrDefault(forOp.getUpperBound()), in hoistPaddingOnTensors()
468 bvm.lookupOrDefault(forOp.getStep()), packedTensor); in hoistPaddingOnTensors()
470 bvm.map(forOp.getInductionVar(), clonedForOp.getInductionVar()); in hoistPaddingOnTensors()
471 bvm.map(forOp.getRegionIterArgs(), clonedForOp.getRegionIterArgs()); in hoistPaddingOnTensors()
472 bvm.map(forOp.getResults(), clonedForOp.getResults()); in hoistPaddingOnTensors()
518 auto forOp = scf::getForInductionVarOwner(iv); in hoistPaddingOnTensors() local
519 b.setInsertionPointToEnd(&forOp.getRegion().front()); in hoistPaddingOnTensors()
521 valueToYield = forOp.getResult(0); in hoistPaddingOnTensors()