| /llvm-project-15.0.7/mlir/lib/Dialect/Async/Transforms/ |
| H A D | AsyncParallelFor.cpp | 376 ImplicitLocOpBuilder b(loc, nestedBuilder); in createParallelComputeFunction()
606 auto syncDispatch = [&](OpBuilder &nestedBuilder, Location loc) { in doAsyncDispatch() argument
607 ImplicitLocOpBuilder b(loc, nestedBuilder); in doAsyncDispatch()
619 auto asyncDispatch = [&](OpBuilder &nestedBuilder, Location loc) { in doAsyncDispatch() argument
620 ImplicitLocOpBuilder b(loc, nestedBuilder); in doAsyncDispatch()
757 auto noOp = [&](OpBuilder &nestedBuilder, Location loc) { in matchAndRewrite() argument
758 nestedBuilder.create<scf::YieldOp>(loc); in matchAndRewrite()
763 auto dispatch = [&](OpBuilder &nestedBuilder, Location loc) { in matchAndRewrite() argument
764 ImplicitLocOpBuilder b(loc, nestedBuilder); in matchAndRewrite()
877 ImplicitLocOpBuilder b(loc, nestedBuilder); in matchAndRewrite()
[all …]
|
| /llvm-project-15.0.7/mlir/lib/Conversion/TosaToLinalg/ |
| H A D | TosaToLinalgNamed.cpp | 294 [&](OpBuilder &nestedBuilder, Location nestedLoc, in matchAndRewrite()
296 Value added = nestedBuilder.create<arith::AddIOp>( in matchAndRewrite()
316 [&](OpBuilder &nestedBuilder, Location nestedLoc, in matchAndRewrite() argument
318 Value added = nestedBuilder.create<arith::AddFOp>( in matchAndRewrite()
457 [&](OpBuilder &nestedBuilder, Location nestedLoc, in matchAndRewrite()
459 Value added = nestedBuilder.create<arith::AddFOp>( in matchAndRewrite()
484 [&](OpBuilder &nestedBuilder, Location nestedLoc, in matchAndRewrite()
486 Value added = nestedBuilder.create<arith::AddIOp>( in matchAndRewrite()
642 [&](OpBuilder &nestedBuilder, Location nestedLoc, in matchAndRewrite()
670 [&](OpBuilder &nestedBuilder, Location nestedLoc, in matchAndRewrite()
[all …]
|
| H A D | TosaToLinalg.cpp | 848 nestedBuilder.create<linalg::YieldOp>(loc, result); in reduceMatchAndRewriteHelper()
1250 [&](OpBuilder &nestedBuilder, Location nestedLoc, in matchAndRewrite()
1262 nestedBuilder); in matchAndRewrite()
1264 op, "output_zp", nestedBuilder.getI32Type(), nestedBuilder); in matchAndRewrite()
1272 value = nestedBuilder in matchAndRewrite()
1292 nestedBuilder.getBoolAttr(doubleRound)); in matchAndRewrite()
1319 nestedBuilder); in matchAndRewrite()
1322 value = nestedBuilder.create<arith::TruncIOp>( in matchAndRewrite()
1327 value = nestedBuilder in matchAndRewrite()
2002 [&](OpBuilder &nestedBuilder, Location nestedLoc, in matchAndRewrite()
[all …]
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch7/mlir/ |
| H A D | LowerToAffineLoops.cpp | 81 [&](OpBuilder &nestedBuilder, Location loc, ValueRange ivs) { in lowerOpToLoops() argument
85 Value valueToStore = processIteration(nestedBuilder, operands, ivs); in lowerOpToLoops()
86 nestedBuilder.create<AffineStoreOp>(loc, valueToStore, alloc, ivs); in lowerOpToLoops()
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch6/mlir/ |
| H A D | LowerToAffineLoops.cpp | 81 [&](OpBuilder &nestedBuilder, Location loc, ValueRange ivs) { in lowerOpToLoops() argument
85 Value valueToStore = processIteration(nestedBuilder, operands, ivs); in lowerOpToLoops()
86 nestedBuilder.create<AffineStoreOp>(loc, valueToStore, alloc, ivs); in lowerOpToLoops()
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch5/mlir/ |
| H A D | LowerToAffineLoops.cpp | 81 [&](OpBuilder &nestedBuilder, Location loc, ValueRange ivs) { in lowerOpToLoops() argument
85 Value valueToStore = processIteration(nestedBuilder, operands, ivs); in lowerOpToLoops()
86 nestedBuilder.create<AffineStoreOp>(loc, valueToStore, alloc, ivs); in lowerOpToLoops()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Utils/ |
| H A D | Utils.cpp | 653 [&](OpBuilder &nestedBuilder, Location nestedLoc, ValueRange localIvs) { in __anon19d834b50e02() argument
656 nestedBuilder, nestedLoc, lbs.drop_front(nOuterPar), in __anon19d834b50e02()
682 [&](OpBuilder &nestedBuilder, Location nestedLoc, ValueRange localIvs) { in __anon19d834b50f02() argument
685 nestedBuilder, nestedLoc, lbs.drop_front(numProcessed), in __anon19d834b50f02()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Transforms.cpp | 725 [&](OpBuilder &nestedBuilder, Location nestedLoc, ValueRange args) { in matchAndRewrite() argument
726 nestedBuilder.create<linalg::YieldOp>(nestedLoc, args[0]); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SCF/IR/ |
| H A D | SCF.cpp | 538 [&](OpBuilder &nestedBuilder, Location nestedLoc, Value iv, in buildLoopNest()
584 [&bodyBuilder](OpBuilder &nestedBuilder, in buildLoopNest()
588 bodyBuilder(nestedBuilder, nestedLoc, ivs); in buildLoopNest()
2139 auto wrappedBuilderFn = [&bodyBuilderFn](OpBuilder &nestedBuilder, in build()
2142 bodyBuilderFn(nestedBuilder, nestedLoc, ivs); in build()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/IR/ |
| H A D | AffineOps.cpp | 2177 auto loopBody = [&](OpBuilder &nestedBuilder, Location nestedLoc, Value iv, in buildAffineLoopNestImpl()
2182 OpBuilder::InsertionGuard nestedGuard(nestedBuilder); in buildAffineLoopNestImpl()
2183 bodyBuilderFn(nestedBuilder, nestedLoc, ivs); in buildAffineLoopNestImpl()
2185 nestedBuilder.create<AffineYieldOp>(nestedLoc); in buildAffineLoopNestImpl()
|