| /external/llvm-project/mlir/lib/Dialect/Linalg/Utils/ |
| D | Utils.cpp | 129 function_ref<scf::ValueVector(ValueRange, ValueRange)> bodyBuilderFn, in doit() argument
141 edsc::loopNestBuilder(lbs, ubs, steps, iterArgInitValues, bodyBuilderFn); in doit()
159 function_ref<scf::ValueVector(ValueRange, ValueRange)> bodyBuilderFn, in doit() argument
175 bodyBuilderFn(ivs, {}); in doit()
205 function_ref<void(ValueRange)> bodyBuilderFn, in generateParallelLoopNest() argument
215 return bodyBuilderFn(ivStorage);
230 bodyBuilderFn, ivStorage, distributionMethod); in __anon7ecdf6bc0202()
246 bodyBuilderFn, ivStorage, in __anon7ecdf6bc0302()
276 iteratorTypes.drop_front(numProcessed), bodyBuilderFn, ivStorage, in __anon7ecdf6bc0402()
295 iteratorTypes.drop_front(numProcessed), bodyBuilderFn, ivStorage, in __anon7ecdf6bc0502()
[all …]
|
| /external/llvm-project/mlir/lib/Dialect/Affine/EDSC/ |
| D | Builders.cpp | 19 function_ref<void(ValueRange)> bodyBuilderFn) { in affineLoopNestBuilder() argument
27 bodyBuilderFn(ivs); in affineLoopNestBuilder()
30 if (bodyBuilderFn) in affineLoopNestBuilder()
40 function_ref<void(Value)> bodyBuilderFn) { in affineLoopBuilder() argument
53 if (bodyBuilderFn) { in affineLoopBuilder()
56 bodyBuilderFn(iv); in affineLoopBuilder()
64 function_ref<void(Value, ValueRange)> bodyBuilderFn) { in affineLoopBuilder() argument
77 if (bodyBuilderFn) { in affineLoopBuilder()
80 bodyBuilderFn(iv, itrArgs); in affineLoopBuilder()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Affine/EDSC/ |
| D | Builders.h | 37 function_ref<void(ValueRange)> bodyBuilderFn = nullptr);
48 function_ref<void(Value)> bodyBuilderFn = nullptr);
61 function_ref<void(Value, ValueRange)> bodyBuilderFn = nullptr);
|
| /external/llvm-project/mlir/lib/Dialect/SCF/ |
| D | SCF.cpp | 808 bodyBuilderFn) { in build() argument
827 if (bodyBuilderFn) { in build()
829 bodyBuilderFn(builder, result.location, in build()
839 function_ref<void(OpBuilder &, Location, ValueRange)> bodyBuilderFn) { in build() argument
843 auto wrappedBuilderFn = [&bodyBuilderFn](OpBuilder &nestedBuilder, in build()
846 bodyBuilderFn(nestedBuilder, nestedLoc, ivs); in build()
849 if (bodyBuilderFn) in build()
1108 function_ref<void(OpBuilder &, Location, Value, Value)> bodyBuilderFn) { in build() argument
1115 if (bodyBuilderFn) in build()
1116 bodyBuilderFn(builder, result.location, body->getArgument(0), in build()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Affine/IR/ |
| D | AffineOps.h | 402 bodyBuilderFn = nullptr);
406 bodyBuilderFn = nullptr);
|
| /external/llvm-project/mlir/include/mlir/Dialect/Linalg/Utils/ |
| D | Utils.h | 221 function_ref<scf::ValueVector(ValueRange, ValueRange)> bodyBuilderFn,
|
| /external/llvm-project/mlir/lib/Dialect/Affine/IR/ |
| D | AffineOps.cpp | 1802 function_ref<void(OpBuilder &, Location, ValueRange)> bodyBuilderFn, in buildAffineLoopNestImpl() argument
1810 if (bodyBuilderFn) in buildAffineLoopNestImpl()
1811 bodyBuilderFn(builder, loc, ValueRange()); in buildAffineLoopNestImpl()
1824 if (i == e - 1 && bodyBuilderFn) { in buildAffineLoopNestImpl()
1826 bodyBuilderFn(nestedBuilder, nestedLoc, ivs); in buildAffineLoopNestImpl()
1842 AffineForOp::BodyBuilderFn bodyBuilderFn) { in buildAffineLoopFromConstants() argument
1844 bodyBuilderFn); in buildAffineLoopFromConstants()
1851 AffineForOp::BodyBuilderFn bodyBuilderFn) { in buildAffineLoopFromValues() argument
1857 bodyBuilderFn); in buildAffineLoopFromValues()
1860 /*iterArgs=*/llvm::None, bodyBuilderFn); in buildAffineLoopFromValues()
[all …]
|
| /external/llvm-project/mlir/include/mlir/Dialect/SCF/ |
| D | SCFOps.td | 358 "nullptr">:$bodyBuilderFn)>,
362 "nullptr">:$bodyBuilderFn)>,
417 "nullptr">:$bodyBuilderFn)>
|