Home
last modified time | relevance | path

Searched refs:loopIvs (Results 1 – 8 of 8) sorted by relevance

/external/llvm-project/mlir/examples/toy/Ch7/mlir/
DLowerToAffineLoops.cpp58 OpBuilder &rewriter, ValueRange memRefOperands, ValueRange loopIvs)>;
107 ValueRange loopIvs) { in matchAndRewrite() argument
116 builder.create<AffineLoadOp>(loc, binaryAdaptor.lhs(), loopIvs); in matchAndRewrite()
118 builder.create<AffineLoadOp>(loc, binaryAdaptor.rhs(), loopIvs); in matchAndRewrite()
230 ValueRange loopIvs) { in matchAndRewrite() argument
239 SmallVector<Value, 2> reverseIvs(llvm::reverse(loopIvs)); in matchAndRewrite()
DLowerToLLVM.cpp70 SmallVector<Value, 4> loopIvs; in matchAndRewrite() local
79 loopIvs.push_back(loop.getInductionVar()); in matchAndRewrite()
94 auto elementLoad = rewriter.create<LoadOp>(loc, printOp.input(), loopIvs); in matchAndRewrite()
/external/llvm-project/mlir/examples/toy/Ch6/mlir/
DLowerToAffineLoops.cpp58 OpBuilder &builder, ValueRange memRefOperands, ValueRange loopIvs)>;
107 ValueRange loopIvs) { in matchAndRewrite() argument
116 builder.create<AffineLoadOp>(loc, binaryAdaptor.lhs(), loopIvs); in matchAndRewrite()
118 builder.create<AffineLoadOp>(loc, binaryAdaptor.rhs(), loopIvs); in matchAndRewrite()
229 ValueRange loopIvs) { in matchAndRewrite() argument
238 SmallVector<Value, 2> reverseIvs(llvm::reverse(loopIvs)); in matchAndRewrite()
DLowerToLLVM.cpp70 SmallVector<Value, 4> loopIvs; in matchAndRewrite() local
79 loopIvs.push_back(loop.getInductionVar()); in matchAndRewrite()
94 auto elementLoad = rewriter.create<LoadOp>(loc, printOp.input(), loopIvs); in matchAndRewrite()
/external/llvm-project/mlir/examples/toy/Ch5/mlir/
DLowerToAffineLoops.cpp58 OpBuilder &rewriter, ValueRange memRefOperands, ValueRange loopIvs)>;
107 ValueRange loopIvs) { in matchAndRewrite() argument
116 builder.create<AffineLoadOp>(loc, binaryAdaptor.lhs(), loopIvs); in matchAndRewrite()
118 builder.create<AffineLoadOp>(loc, binaryAdaptor.rhs(), loopIvs); in matchAndRewrite()
230 ValueRange loopIvs) { in matchAndRewrite() argument
239 SmallVector<Value, 2> reverseIvs(llvm::reverse(loopIvs)); in matchAndRewrite()
/external/llvm-project/mlir/lib/Dialect/GPU/Transforms/
DMemoryPromotion.cpp82 loopNestBuilder(lbs, ubs, steps, [&](ValueRange loopIvs) { in insertCopyLoops() argument
83 ivs.assign(loopIvs.begin(), loopIvs.end()); in insertCopyLoops()
/external/llvm-project/mlir/lib/Conversion/VectorToSCF/
DVectorToSCF.cpp580 loopNestBuilder(lbs, ubs, steps, [&](ValueRange loopIvs) { in matchAndRewrite() argument
581 auto ivsStorage = llvm::to_vector<8>(loopIvs); in matchAndRewrite()
666 loopNestBuilder(lbs, ubs, steps, [&](ValueRange loopIvs) { in matchAndRewrite() argument
667 auto ivsStorage = llvm::to_vector<8>(loopIvs); in matchAndRewrite()
/external/llvm-project/mlir/docs/Tutorials/Toy/
DCh-5.md122 ArrayRef<mlir::Value> loopIvs) {
132 SmallVector<mlir::Value, 2> reverseIvs(llvm::reverse(loopIvs));