| /llvm-project/mlir/lib/Dialect/Affine/Transforms/ |
| H A D | AffineLoopInvariantCodeMotion.cpp | 78 isOpLoopInvariant(Operation & op,Value indVar,ValueRange iterArgs,SmallPtrSetImpl<Operation * > & opsWithUsers,SmallPtrSetImpl<Operation * > & opsToHoist) isOpLoopInvariant() argument
174 areAllOpsInTheBlockListInvariant(Region & blockList,Value indVar,ValueRange iterArgs,SmallPtrSetImpl<Operation * > & opsWithUsers,SmallPtrSetImpl<Operation * > & opsToHoist) areAllOpsInTheBlockListInvariant() argument
190 checkInvarianceOfNestedIfOps(AffineIfOp ifOp,Value indVar,ValueRange iterArgs,SmallPtrSetImpl<Operation * > & opsWithUsers,SmallPtrSetImpl<Operation * > & opsToHoist) checkInvarianceOfNestedIfOps() argument
207 ValueRange iterArgs = forOp.getRegionIterArgs(); runOnAffineForOp() local
|
| /llvm-project/mlir/lib/Dialect/Bufferization/Transforms/ |
| H A D | LowerDeallocations.cpp | 458 __anon5b7691f10802(OpBuilder &builder, Location loc, Value i, ValueRange iterArgs) buildDeallocationLibraryFunction() argument
466 __anon5b7691f10902(OpBuilder &builder, Location loc, Value outerIter, ValueRange iterArgs) buildDeallocationLibraryFunction() argument
479 __anon5b7691f10a02(OpBuilder &builder, Location loc, Value i, ValueRange iterArgs) buildDeallocationLibraryFunction() argument
515 __anon5b7691f10c02(OpBuilder &builder, Location loc, Value i, ValueRange iterArgs) buildDeallocationLibraryFunction() argument
|
| /llvm-project/mlir/lib/Conversion/SCFToControlFlow/ |
| H A D | SCFToControlFlow.cpp | 482 SmallVector<Value, 4> iterArgs = llvm::to_vector<4>(parallelOp.getInitVals()); matchAndRewrite() local
|
| /llvm-project/mlir/test/lib/Dialect/Tensor/ |
| H A D | TestTensorTransforms.cpp | 226 __anon0043a2a40402(OpBuilder &nestedBuilder, Location loc, ValueRange outputIvs, ValueRange iterArgs) emitReplacement() argument
|
| /llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/Utils/ |
| H A D | SparseTensorIterator.cpp | 1224 SmallVector<Value> iterArgs; in inflateSubSectTree() local
1230 ValueRange iterArgs) { in inflateSubSectTree()
1277 [&](OpBuilder &b, Location l, Value crd, ValueRange iterArgs) { in inflateSubSectTree()
1442 ValueRange iterArgs) -> scf::ValueVector { in forwardImpl()
|
| /llvm-project/mlir/lib/Dialect/SCF/Utils/ |
| H A D | Utils.cpp | 323 generateUnrolledLoop(Block * loopBodyBlock,Value forOpIV,uint64_t unrollFactor,function_ref<Value (unsigned,Value,OpBuilder)> ivRemapFn,function_ref<void (unsigned,Operation *,OpBuilder)> annotateFn,ValueRange iterArgs,ValueRange yieldedValues) generateUnrolledLoop() argument
471 auto iterArgs = ValueRange(forOp.getRegionIterArgs()); loopUnrollByFactor() local
|
| /llvm-project/mlir/lib/Dialect/SparseTensor/IR/ |
| H A D | SparseTensorDialect.cpp | 2137 parseSparseSpaceLoop(OpAsmParser & parser,OperationState & state,SmallVectorImpl<OpAsmParser::Argument> & iterators,SmallVectorImpl<OpAsmParser::Argument> & iterArgs) parseSparseSpaceLoop() argument
2342 SmallVector<OpAsmParser::Argument> iters, iterArgs; parse() local
2436 auto iterArgs = getRegionIterArgs(); verifyRegions() local
[all...] |
| /llvm-project/mlir/lib/Dialect/Affine/Utils/ |
| H A D | LoopUtils.cpp | 107 auto iterArgs = forOp.getRegionIterArgs(); replaceIterArgsAndYieldResults() local
914 generateUnrolledLoop(Block * loopBodyBlock,Value forOpIV,uint64_t unrollFactor,function_ref<Value (unsigned,Value,OpBuilder)> ivRemapFn,function_ref<void (unsigned,Operation *,OpBuilder)> annotateFn,ValueRange iterArgs,ValueRange yieldedValues) generateUnrolledLoop() argument
1050 ValueRange iterArgs(forOp.getRegionIterArgs()); loopUnrollByFactor() local
[all...] |
| /llvm-project/mlir/lib/Conversion/ShapeToStandard/ |
| H A D | ShapeToStandard.cpp | 277 [&](OpBuilder &b, Location loc, Value iv, ValueRange iterArgs) { in matchAndRewrite()
|
| /llvm-project/flang/lib/Optimizer/Dialect/ |
| H A D | FIROps.cpp | 2134 build(mlir::OpBuilder & builder,mlir::OperationState & result,mlir::Value lb,mlir::Value ub,mlir::Value step,mlir::Value iterate,bool finalCountValue,mlir::ValueRange iterArgs,llvm::ArrayRef<mlir::NamedAttribute> attributes) build() argument
2290 auto iterArgs = getRegionIterArgs(); verify() local
2458 build(mlir::OpBuilder & builder,mlir::OperationState & result,mlir::Value lb,mlir::Value ub,mlir::Value step,bool unordered,bool finalCountValue,mlir::ValueRange iterArgs,mlir::ValueRange reduceOperands,llvm::ArrayRef<mlir::Attribute> reduceAttrs,llvm::ArrayRef<mlir::NamedAttribute> attributes) build() argument
2639 auto iterArgs = getRegionIterArgs(); verify() local
[all...] |
| /llvm-project/mlir/lib/Dialect/Affine/IR/ |
| H A D | AffineOps.cpp | 1813 build(OpBuilder & builder,OperationState & result,ValueRange lbOperands,AffineMap lbMap,ValueRange ubOperands,AffineMap ubMap,int64_t step,ValueRange iterArgs,BodyBuilderFn bodyBuilder) build() argument
1872 build(OpBuilder & builder,OperationState & result,int64_t lb,int64_t ub,int64_t step,ValueRange iterArgs,BodyBuilderFn bodyBuilder) build() argument
2291 auto iterArgs = forOp.getRegionIterArgs(); matchAndRewrite() local
2632 __anon41ba865e1602(OpBuilder &nestedBuilder, Location nestedLoc, Value iv, ValueRange iterArgs) buildAffineLoopNestImpl() argument
[all...] |
| /llvm-project/mlir/lib/Dialect/SCF/IR/ |
| H A D | SCF.cpp | 364 auto iterArgs = getRegionIterArgs(); in verifyRegions() local
689 ValueRange steps, ValueRange iterArgs, in buildLoopNest() argument
315 build(OpBuilder & builder,OperationState & result,Value lb,Value ub,Value step,ValueRange iterArgs,BodyBuilderFn bodyBuilder) build() argument
|
| /llvm-project/mlir/lib/Dialect/SCF/Transforms/ |
| H A D | BufferizableOpInterfaceImpl.cpp | 764 SmallVector<Value> iterArgs = bufferize() local
|
| H A D | TileUsingInterface.cpp | 483 __anone7fd0eaf0702(OpBuilder &b, Location loc, Value iv, ValueRange iterArgs) addInitOperandsToLoopNest() argument
|
| /llvm-project/mlir/lib/Dialect/Linalg/Utils/ |
| H A D | Utils.cpp | 330 __anon42121b780702(OpBuilder &b, Location loc, ValueRange ivs, ValueRange iterArgs) doit() argument
|