Searched refs:loopNest (Results 1 – 5 of 5) sorted by relevance
| /llvm-project-15.0.7/mlir/test/lib/Dialect/Affine/ |
| H A D | TestAffineDataCopy.cpp | 71 auto loopNest = depthToLoops[0][0]; in runOnOperation() local
94 if (failed(affineDataCopyGenerate(loopNest, copyOptions, load.getMemRef(), in runOnOperation()
99 MemRefRegion region(loopNest.getLoc()); in runOnOperation()
102 if (failed(generateCopyForMemRegion(region, loopNest, copyOptions, result))) in runOnOperation()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SCF/Utils/ |
| H A D | Utils.cpp | 106 OpBuilder &builder, ArrayRef<scf::ForOp> loopNest, in replaceLoopNestWithNewYields() argument
108 if (loopNest.empty()) in replaceLoopNestWithNewYields()
110 SmallVector<scf::ForOp> newLoopNest(loopNest.size()); in replaceLoopNestWithNewYields()
113 builder, loopNest.back(), newIterOperands, newYieldValueFn); in replaceLoopNestWithNewYields()
116 llvm::reverse(llvm::seq<unsigned>(0, loopNest.size() - 1))) { in replaceLoopNestWithNewYields()
125 builder, loopNest[loopDepth], newIterOperands, fn); in replaceLoopNestWithNewYields()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Tiling.cpp | 549 tensor::PadOp &newPadOp, LoopNest &loopNest, in tilePadOp() argument
580 loopNest = mlir::scf::buildLoopNest( in tilePadOp()
617 LoopNest loopNest; in matchAndRewrite() local
618 if (failed(tilePadOp(rewriter, op, newPadOp, loopNest, options))) in matchAndRewrite()
623 rewriter.replaceOp(op, loopNest.getResults()[0]); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/SCF/Utils/ |
| H A D | Utils.h | 70 replaceLoopNestWithNewYields(OpBuilder &builder, ArrayRef<scf::ForOp> loopNest,
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Utils/ |
| H A D | Utils.cpp | 528 LoopNest loopNest = mlir::scf::buildLoopNest( in doit() local
542 if (!distributionOptions || loopNest.loops.empty()) in doit()
549 loops.push_back(loopNest.loops[iteratorType.index()]); in doit()
|