| /llvm-project/mlir/lib/Dialect/Linalg/Utils/ |
| H A D | Utils.cpp | 318 doit(OpBuilder & b,Location loc,ArrayRef<Range> loopRanges,LinalgOp linalgOp,ArrayRef<utils::IteratorType> iteratorTypes,function_ref<scf::ValueVector (OpBuilder &,Location,ValueRange,ValueRange)> bodyBuilderFn,ArrayRef<linalg::ProcInfo> procInfo) doit() argument
361 doit(OpBuilder & b,Location loc,ArrayRef<Range> loopRanges,LinalgOp linalgOp,ArrayRef<utils::IteratorType> iteratorTypes,function_ref<scf::ValueVector (OpBuilder &,Location,ValueRange,ValueRange)> bodyBuilderFn,ArrayRef<linalg::ProcInfo>) doit() argument
412 generateParallelLoopNest(OpBuilder & b,Location loc,ValueRange lbs,ValueRange ubs,ValueRange steps,ArrayRef<utils::IteratorType> iteratorTypes,ArrayRef<linalg::ProcInfo> procInfo,function_ref<void (OpBuilder &,Location,ValueRange)> bodyBuilderFn,SmallVectorImpl<Value> & ivStorage) generateParallelLoopNest() argument
528 doit(OpBuilder & b,Location loc,ArrayRef<Range> loopRanges,LinalgOp linalgOp,ArrayRef<utils::IteratorType> iteratorTypes,function_ref<scf::ValueVector (OpBuilder &,Location,ValueRange,ValueRange)> bodyBuilderFn,ArrayRef<linalg::ProcInfo> procInfo) doit() argument
[all...] |
| /llvm-project/mlir/lib/Dialect/Affine/IR/ |
| H A D | AffineOps.cpp | 2676 /*iterArgs=*/std::nullopt, bodyBuilderFn); in buildAffineLoopNest() argument
2613 buildAffineLoopNestImpl(OpBuilder & builder,Location loc,BoundListTy lbs,BoundListTy ubs,ArrayRef<int64_t> steps,function_ref<void (OpBuilder &,Location,ValueRange)> bodyBuilderFn,LoopCreatorTy && loopCreatorFn) buildAffineLoopNestImpl() argument
2653 buildAffineLoopFromConstants(OpBuilder & builder,Location loc,int64_t lb,int64_t ub,int64_t step,AffineForOp::BodyBuilderFn bodyBuilderFn) buildAffineLoopFromConstants() argument
2662 buildAffineLoopFromValues(OpBuilder & builder,Location loc,Value lb,Value ub,int64_t step,AffineForOp::BodyBuilderFn bodyBuilderFn) buildAffineLoopFromValues() argument
2684 buildAffineLoopNest(OpBuilder & builder,Location loc,ValueRange lbs,ValueRange ubs,ArrayRef<int64_t> steps,function_ref<void (OpBuilder &,Location,ValueRange)> bodyBuilderFn) buildAffineLoopNest() argument
[all...] |
| /llvm-project/mlir/lib/Dialect/SCF/IR/ |
| H A D | SCF.cpp | 1327 build(mlir::OpBuilder & b,mlir::OperationState & result,ArrayRef<OpFoldResult> lbs,ArrayRef<OpFoldResult> ubs,ArrayRef<OpFoldResult> steps,ValueRange outputs,std::optional<ArrayAttr> mapping,function_ref<void (OpBuilder &,Location,ValueRange)> bodyBuilderFn) build() argument
1383 build(mlir::OpBuilder & b,mlir::OperationState & result,ArrayRef<OpFoldResult> ubs,ValueRange outputs,std::optional<ArrayAttr> mapping,function_ref<void (OpBuilder &,Location,ValueRange)> bodyBuilderFn) build() argument
2816 build(OpBuilder & builder,OperationState & result,ValueRange lowerBounds,ValueRange upperBounds,ValueRange steps,ValueRange initVals,function_ref<void (OpBuilder &,Location,ValueRange,ValueRange)> bodyBuilderFn) build() argument
2850 build(OpBuilder & builder,OperationState & result,ValueRange lowerBounds,ValueRange upperBounds,ValueRange steps,function_ref<void (OpBuilder &,Location,ValueRange)> bodyBuilderFn) build() argument
[all...] |