| /external/llvm-project/mlir/lib/Conversion/SCFToStandard/ |
| D | SCFToStandard.cpp | 413 SmallVector<Value, 4> iterArgs = llvm::to_vector<4>(parallelOp.initVals()); in matchAndRewrite() local
417 SmallVector<Value, 4> loopResults(iterArgs); in matchAndRewrite()
423 ForOp forOp = rewriter.create<ForOp>(loc, lower, upper, step, iterArgs); in matchAndRewrite()
426 iterArgs.assign(iterRange.begin(), iterRange.end()); in matchAndRewrite()
452 Value arg = iterArgs[yieldOperands.size()]; in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Dialect/SCF/ |
| D | SCF.cpp | 75 Value ub, Value step, ValueRange iterArgs, in build() argument
78 result.addOperands(iterArgs); in build()
79 for (Value v : iterArgs) in build()
85 for (Value v : iterArgs) in build()
91 if (iterArgs.empty() && !bodyBuilder) { in build()
127 auto iterArgs = op.getRegionIterArgs(); in verify() local
130 for (auto e : llvm::zip(iterOperands, iterArgs, opResults)) { in verify()
310 ValueRange steps, ValueRange iterArgs, in buildLoopNest() argument
321 bodyBuilder ? bodyBuilder(builder, loc, ValueRange(), iterArgs) in buildLoopNest()
323 assert(results.size() == iterArgs.size() && in buildLoopNest()
[all …]
|
| /external/llvm-project/mlir/include/mlir/Dialect/SCF/ |
| D | SCF.h | 80 ValueRange steps, ValueRange iterArgs,
|
| D | SCFOps.td | 158 CArg<"ValueRange", "llvm::None">:$iterArgs,
|
| /external/llvm-project/flang/lib/Optimizer/Dialect/ |
| D | FIROps.cpp | 603 mlir::Value iterate, mlir::ValueRange iterArgs, in build() argument
607 result.addOperands(iterArgs); in build()
608 for (auto v : iterArgs) in build()
614 bodyRegion->front().addArguments(iterArgs.getTypes()); in build()
717 auto iterArgs = op.getRegionIterArgs(); in verify() local
720 for (auto e : llvm::zip(iterOperands, iterArgs, opResults)) { in verify()
790 bool unordered, mlir::ValueRange iterArgs, in build() argument
793 result.addOperands(iterArgs); in build()
794 for (auto v : iterArgs) in build()
798 if (iterArgs.empty()) in build()
[all …]
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | Tiling.cpp | 380 [&](ValueRange localIvs, ValueRange iterArgs) -> scf::ValueVector { in tileLinalgOpImpl() argument
394 assert(op.getNumInitTensors() == iterArgs.size() && in tileLinalgOpImpl()
399 std::copy(iterArgs.begin(), iterArgs.end(), in tileLinalgOpImpl()
|
| D | Loops.cpp | 524 [&](ValueRange ivs, ValueRange iterArgs) -> scf::ValueVector { in linalgOpToLoopsImpl() argument
525 assert(iterArgs.empty() && "unexpected iterArgs"); in linalgOpToLoopsImpl()
|
| /external/llvm-project/mlir/lib/Dialect/Affine/EDSC/ |
| D | Builders.cpp | 63 ValueRange lbs, ValueRange ubs, int64_t step, ValueRange iterArgs, in affineLoopBuilder() argument
74 builder.getMultiDimIdentityMap(ubs.size()), step, iterArgs, in affineLoopBuilder()
|
| /external/llvm-project/mlir/lib/Dialect/Affine/IR/ |
| D | AffineOps.cpp | 1213 ValueRange iterArgs, BodyBuilderFn bodyBuilder) { in build() argument
1222 for (Value val : iterArgs) in build()
1237 result.addOperands(iterArgs); in build()
1244 for (Value val : iterArgs) in build()
1250 if (iterArgs.empty() && !bodyBuilder) { in build()
1261 int64_t ub, int64_t step, ValueRange iterArgs, in build() argument
1265 return build(builder, result, {}, lbMap, {}, ubMap, step, iterArgs, in build()
1821 ValueRange iterArgs) { in buildAffineLoopNestImpl() argument
|
| /external/llvm-project/mlir/include/mlir/Dialect/Affine/EDSC/ |
| D | Builders.h | 60 ValueRange lbs, ValueRange ubs, int64_t step, ValueRange iterArgs,
|
| /external/llvm-project/mlir/lib/Conversion/ShapeToStandard/ |
| D | ShapeToStandard.cpp | 264 [&](OpBuilder &b, Location loc, Value iv, ValueRange iterArgs) { in matchAndRewrite() argument
278 loc, iterArgs[0], in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Transforms/Utils/ |
| D | LoopUtils.cpp | 211 auto iterArgs = forOp.getRegionIterArgs(); in promoteIfSingleIteration() local
212 for (auto e : llvm::zip(iterOperands, iterArgs)) in promoteIfSingleIteration()
1059 ValueRange iterArgs, ValueRange yieldedValues) { in generateUnrolledLoop() argument
1075 operandMap.map(iterArgs, lastYielded); in generateUnrolledLoop()
1257 auto iterArgs = ValueRange(forOp.getRegionIterArgs()); in loopUnrollByFactor() local
1268 iterArgs, yieldedValues); in loopUnrollByFactor()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Affine/IR/ |
| D | AffineOps.td | 217 CArg<"int64_t", "1">:$step, CArg<"ValueRange", "llvm::None">:$iterArgs,
222 CArg<"ValueRange", "llvm::None">:$iterArgs,
|
| /external/llvm-project/flang/include/flang/Optimizer/Dialect/ |
| D | FIROps.td | 1925 CArg<"ValueRange", "llvm::None">:$iterArgs,
2047 CArg<"ValueRange", "llvm::None">:$iterArgs,
|