Home
last modified time | relevance | path

Searched defs:ivs (Results 1 – 25 of 31) sorted by relevance

12

/llvm-project/mlir/lib/Dialect/SCF/Transforms/
H A DParallelLoopTiling.cpp168 for (const auto &ivs : llvm::enumerate(llvm::zip( in tileParallelLoop() local
179 for (auto ivs : llvm::zip(innerLoop.getInductionVars(), in tileParallelLoop() local
H A DForallToFor.cpp42 SmallVector<Value> ivs = llvm::map_to_vector( forallToForLoop() local
H A DTileUsingInterface.cpp155 SmallVector<Value> ivs; generateLoopNestUsingForOp() local
470 SmallVector<Value> ivs; addInitOperandsToLoopNest() local
583 __anone7fd0eaf0802(RewriterBase &rewriter, Location loc, ValueRange ivs, ValueRange regionIterArgs, SmallVector<Value> &tiledResults, SmallVector<SmallVector<OpFoldResult>> &resultOffsets, SmallVector<SmallVector<OpFoldResult>> &resultSizes) tileUsingSCF() argument
727 __anone7fd0eaf0a02(RewriterBase &rewriter, Location loc, ValueRange ivs, ValueRange regionIterArgs, SmallVector<Value> &tiledResult, SmallVector<SmallVector<OpFoldResult>> &resultOffsets, SmallVector<SmallVector<OpFoldResult>> &resultSizes) tileReductionUsingScf() argument
1616 SmallVector<Value> ivs; lowerToLoopsUsingSCFForOp() local
[all...]
/llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparseIterationToScf.cpp95 SmallVector<Value> ivs; matchAndRewrite() local
122 SmallVector<Value> ivs; matchAndRewrite() local
H A DSparsification.cpp407 genConditionalInsert(Location loc,OpBuilder & builder,Value cond,Value sparseOut,ValueRange ivs,Value v) genConditionalInsert() argument
431 SmallVector<Value> ivs = llvm::to_vector(llvm::drop_end( genInsertionStore() local
/llvm-project/mlir/lib/Dialect/Linalg/Utils/
H A DUtils.cpp330 __anon42121b780702(OpBuilder &b, Location loc, ValueRange ivs, ValueRange iterArgs) doit() argument
380 __anon42121b780802(OpBuilder &b, Location loc, ValueRange ivs) doit() argument
431 __anon42121b780902(OpBuilder &b, Location loc, ValueRange ivs) generateParallelLoopNest() argument
540 SmallVector<Value, 8> lbsStorage, ubsStorage, stepsStorage, ivs; doit() local
561 __anon42121b780e02(OpBuilder &b, Location loc, ValueRange ivs) doit() argument
711 computeTileOffsets(OpBuilder & b,Location loc,ArrayRef<OpFoldResult> ivs,ArrayRef<OpFoldResult> tileSizes) computeTileOffsets() argument
779 computeAllSliceParameters(OpBuilder & builder,Location loc,LinalgOp linalgOp,ValueRange valuesToTile,ArrayRef<OpFoldResult> ivs,ArrayRef<OpFoldResult> tileSizes,ArrayRef<OpFoldResult> sizeBounds,bool omitPartialTileCheck) computeAllSliceParameters() argument
831 makeTiledShapes(OpBuilder & builder,Location loc,LinalgOp linalgOp,ValueRange valuesToTile,ArrayRef<OpFoldResult> ivs,ArrayRef<OpFoldResult> tileSizes,ArrayRef<OpFoldResult> sizeBounds,bool omitPartialTileCheck) makeTiledShapes() argument
[all...]
/llvm-project/mlir/lib/Dialect/GPU/Transforms/
H A DMemoryPromotion.cpp67 SmallVector<Value, 8> ivs(lbs.size()); in insertCopyLoops() local
/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DTilingInterfaceImpl.cpp34 ValueRange ivs) { in getIndicesForAccess() argument
49 ValueRange ivs, ValueRange argValues) { in inlinePayload() argument
H A DFusion.cpp115 SmallVector<OpFoldResult> ivs, tileSizes, sizeBounds; in fuse() local
H A DLoops.cpp230 __anon3d41e3320602(OpBuilder &b, Location loc, ValueRange ivs, ValueRange operandValuesToUse) linalgOpToLoopsImpl() argument
H A DTiling.cpp79 RewriterBase &b, LinalgOp op, SmallVectorImpl<Value> &ivs, in transformIndexOps() argument
708 SmallVector<Value, 4> ivs, tensorResults; tileLinalgOpImpl() local
[all...]
/llvm-project/mlir/lib/Conversion/TosaToSCF/
H A DTosaToSCF.cpp120 ValueRange args) -> scf::ValueVector { in matchAndRewrite()
/llvm-project/mlir/lib/Dialect/Tensor/TransformOps/
H A DTensorTransformOps.cpp193 SmallVector<Value> ivs; in applyToOne() local
/llvm-project/mlir/include/mlir/Dialect/Affine/Analysis/
H A DUtils.h262 SmallVector<Value, 4> ivs; member
/llvm-project/mlir/examples/toy/Ch5/mlir/
H A DLowerToAffineLoops.cpp97 [&](OpBuilder &nestedBuilder, Location loc, ValueRange ivs) { in lowerOpToLoops() argument
/llvm-project/mlir/examples/toy/Ch6/mlir/
H A DLowerToAffineLoops.cpp97 [&](OpBuilder &nestedBuilder, Location loc, ValueRange ivs) { in lowerOpToLoops() argument
/llvm-project/mlir/examples/toy/Ch7/mlir/
H A DLowerToAffineLoops.cpp97 [&](OpBuilder &nestedBuilder, Location loc, ValueRange ivs) { in lowerOpToLoops() argument
/llvm-project/mlir/lib/Dialect/MemRef/TransformOps/
H A DMemRefTransformOps.cpp269 SmallVector<Value> ivs; in applyToOne() local
/llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/Utils/
H A DSparseTensorIterator.cpp1064 [this, pos](OpBuilder &b, Location l, ValueRange ivs) { in genSegmentHigh()
1082 [](OpBuilder &b, Location l, ValueRange ivs) { in genSegmentHigh()
1146 [this](OpBuilder &b, Location l, ValueRange ivs) { in forwardImpl()
1164 [this](OpBuilder &b, Location l, ValueRange ivs) { in forwardImpl()
1246 [&helper](OpBuilder &b, Location l, ValueRange ivs) { in inflateSubSectTree()
1251 [&helper, &builder](OpBuilder &b, Location l, ValueRange ivs) { in inflateSubSectTree()
1442 ValueRange iterArgs) -> scf::ValueVector { in forwardImpl()
H A DLoopEmitter.cpp531 SmallVector<Value> ivs; emitWhileLoopOverTensorsAtLvls() local
/llvm-project/mlir/lib/Conversion/SCFToControlFlow/
H A DSCFToControlFlow.cpp483 SmallVector<Value, 4> ivs; matchAndRewrite() local
/llvm-project/mlir/lib/Conversion/SCFToGPU/
H A DSCFToGPU.cpp168 SmallVector<Value, 6> ivs; member
/llvm-project/mlir/lib/Dialect/Affine/Analysis/
H A DUtils.cpp1061 SmallVector<Value, 4> ivs; compute() local
1369 addMissingLoopIVBounds(SmallPtrSet<Value,8> & ivs,FlatAffineValueConstraints * cst) addMissingLoopIVBounds() argument
1876 getAffineIVs(Operation & op,SmallVectorImpl<Value> & ivs) getAffineIVs() argument
[all...]
/llvm-project/mlir/lib/Dialect/SCF/IR/
H A DSCF.cpp712 SmallVector<Value, 4> ivs; in buildLoopNest() local
768 __anon200d06d20702(OpBuilder &nestedBuilder, Location nestedLoc, ValueRange ivs, ValueRange) buildLoopNest() argument
1226 SmallVector<OpAsmParser::Argument, 4> ivs; parse() local
2856 __anon200d06d21c02(OpBuilder &nestedBuilder, Location nestedLoc, ValueRange ivs, ValueRange) build() argument
2930 SmallVector<OpAsmParser::Argument, 4> ivs; parse() local
[all...]
/llvm-project/flang/lib/Semantics/
H A Dresolve-directives.cpp1265 __anon5e3236251602(llvm::SmallVector<Symbol *> &ivs, semantics::UnorderedSymbolSet &symbols) CheckAssociatedLoop() argument
1275 llvm::SmallVector<Symbol *> ivs; CheckAssociatedLoop() local
1714 llvm::SmallVector<const parser::Name *> ivs; Pre() local

12