| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | Generalization.cpp | 50 [®ionBuilder](OpBuilder &bodyBuilder, Location loc, ValueRange) { in createGenericOpFromNamedOp() argument
51 edsc::ScopedContext scope(bodyBuilder, loc); in createGenericOpFromNamedOp()
52 regionBuilder(*bodyBuilder.getBlock()); in createGenericOpFromNamedOp()
158 [](OpBuilder &bodyBuilder, Location bodyLoc, ValueRange bodyArgs) { in createGenericOp() argument
160 bodyBuilder.create<MulFOp>(bodyLoc, bodyArgs[0], bodyArgs[1]); in createGenericOp()
161 Value add = bodyBuilder.create<AddFOp>(bodyLoc, mul, bodyArgs[2]); in createGenericOp()
162 bodyBuilder.create<linalg::YieldOp>(bodyLoc, add); in createGenericOp()
|
| /external/llvm-project/mlir/include/mlir/Dialect/SCF/ |
| D | SCF.h | 82 bodyBuilder = nullptr);
91 bodyBuilder = nullptr);
|
| /external/llvm-project/flang/include/flang/Lower/ |
| D | FIRBuilder.h | 113 std::function<void(FirOpBuilder &)> bodyBuilder,
127 std::function<void(FirOpBuilder &)> bodyBuilder,
129 return createGlobal(loc, type, name, /*isConst=*/true, bodyBuilder,
|
| /external/llvm-project/mlir/lib/Dialect/Async/IR/ |
| D | Async.cpp | 146 ValueRange operands, BodyBuilderFn bodyBuilder) { in build() argument
178 if (resultTypes.empty() && !bodyBuilder) { in build()
182 } else if (bodyBuilder) { in build()
185 bodyBuilder(builder, result.location, bodyBlock.getArguments()); in build()
|
| /external/llvm-project/mlir/lib/Dialect/StandardOps/Transforms/ |
| D | ExpandOps.cpp | 59 OpBuilder bodyBuilder = in matchAndRewrite() local
64 Value cmp = bodyBuilder.create<CmpFOp>(loc, predicate, lhs, rhs); in matchAndRewrite()
65 Value select = bodyBuilder.create<SelectOp>(loc, cmp, lhs, rhs); in matchAndRewrite()
66 bodyBuilder.create<AtomicYieldOp>(loc, select); in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Dialect/SCF/ |
| D | SCF.cpp | 76 BodyBuilderFn bodyBuilder) { in build() argument
91 if (iterArgs.empty() && !bodyBuilder) { in build()
93 } else if (bodyBuilder) { in build()
96 bodyBuilder(builder, result.location, bodyBlock.getArgument(0), in build()
312 bodyBuilder) { in buildLoopNest() argument
321 bodyBuilder ? bodyBuilder(builder, loc, ValueRange(), iterArgs) in buildLoopNest()
365 ValueVector results = bodyBuilder in buildLoopNest()
366 ? bodyBuilder(builder, currentLoc, ivs, in buildLoopNest()
384 function_ref<void(OpBuilder &, Location, ValueRange)> bodyBuilder) { in buildLoopNest() argument
387 [&bodyBuilder](OpBuilder &nestedBuilder, in buildLoopNest()
[all …]
|
| /external/llvm-project/flang/lib/Lower/ |
| D | FIRBuilder.cpp | 121 std::function<void(FirOpBuilder &)> bodyBuilder, mlir::StringAttr linkage) { in createGlobal() argument
133 bodyBuilder(*this); in createGlobal()
|
| /external/llvm-project/mlir/lib/Dialect/Affine/IR/ |
| D | AffineOps.cpp | 1213 ValueRange iterArgs, BodyBuilderFn bodyBuilder) { in build() argument
1250 if (iterArgs.empty() && !bodyBuilder) { in build()
1252 } else if (bodyBuilder) { in build()
1255 bodyBuilder(builder, result.location, inductionVar, in build()
1262 BodyBuilderFn bodyBuilder) { in build() argument
1266 bodyBuilder); in build()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Async/IR/ |
| D | AsyncOps.td | 90 "nullptr">:$bodyBuilder)>,
|
| /external/llvm-project/mlir/lib/Transforms/Utils/ |
| D | LoopUtils.cpp | 267 auto bodyBuilder = OpBuilder::atBlockTerminator(loopChunk.getBody()); in generateShiftedLoop() local
277 auto ivRemap = bodyBuilder.create<AffineApplyOp>( in generateShiftedLoop()
279 bodyBuilder.getSingleDimShiftAffineMap( in generateShiftedLoop()
287 bodyBuilder.clone(*op, operandMap); in generateShiftedLoop()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Affine/IR/ |
| D | AffineOps.td | 219 "nullptr">:$bodyBuilder)>,
224 "nullptr">:$bodyBuilder)>
|
| /external/llvm-project/mlir/lib/Dialect/StandardOps/IR/ |
| D | Ops.cpp | 1786 function_ref<void(OpBuilder &, Location, ValueRange)> bodyBuilder) { in build() argument
1796 bodyBuilder(b, result.location, bodyBlock->getArguments()); in build()
|