Searched refs:loopOps (Results 1 – 5 of 5) sorted by relevance
64 SmallVector<SmallVector<Operation *>> loopOps(numLoops); in applyTileAndFuseToAll() local 135 loopOps[i].push_back(tiledResults->loops[i]); in applyTileAndFuseToAll() 140 transformResults.set(transformOp->getOpResult(i + 1), loopOps[i]); in applyTileAndFuseToAll() 233 SmallVector<Operation *> loopOps; in applyTileToAll() 259 loopOps.push_back(loop); in applyTileToAll() 263 for (auto [index, loop] : llvm::enumerate(loopOps)) in applyTileToAll() 307 SmallVector<SmallVector<Operation *>> loopOps(1); in applyTilingToAll() 337 loopOps[0] = {tiledResults->loops[0]}; in applyTilingToAll() 341 if (!loopOps.empty()) in applyTilingToAll() 342 transformResults.set(transformOp->getOpResult(1), loopOps[ in applyTilingToAll() 228 SmallVector<Operation *> loopOps; applyTileToAll() local 305 SmallVector<SmallVector<Operation *>> loopOps(1); applyTilingToAll() local [all...]
181 ArrayRef<Operation *> loopOps) { in replaceIndexOpsByInductionVariables() argument 184 for (Operation *loopOp : loopOps) { in replaceIndexOpsByInductionVariables() 200 if (!loopOps.empty()) { in replaceIndexOpsByInductionVariables() 201 auto loopOp = cast<LoopLikeOpInterface>(loopOps.back()); in replaceIndexOpsByInductionVariables()
245 SmallVector<Operation *, 8> loopOps; in getIndexSet() local254 loopOps.push_back(forOp); in getIndexSet()258 loopOps.push_back(parallelOp); in getIndexSet()262 extractInductionVars(loopOps, indices); in getIndexSet()
463 auto &loopOps = loop.getBody()->getOperations(); in matchAndRewrite() local470 std::prev(affineFor.getBody()->end()), loopOps, loopOps.begin(), in matchAndRewrite()471 std::prev(loopOps.end())); in matchAndRewrite()
523 SmallVector<SmallVector<Operation *>> loopOps(numLoops); in applyTilingToAll() 554 loopOps[i].push_back(tiledResults->loops[i]); in apply() 559 transformResults.set(transformOp->getOpResult(i + 1), loopOps[i]); in apply() 1980 auto loopOps = state.getPayloadOps(getLoop()); in verify() 1981 if (!llvm::hasSingleElement(targetOps) || !llvm::hasSingleElement(loopOps)) { in verify() 1985 << llvm::range_size(loopOps) << ")"; in verify() 1989 auto loopOp = dyn_cast_or_null<scf::ForOp>(*loopOps.begin());502 SmallVector<SmallVector<Operation *>> loopOps(numLoops); applyTilingToAll() local 1948 auto loopOps = state.getPayloadOps(getLoop()); apply() local