Searched defs:tripCounts (Results 1 – 3 of 3) sorted by relevance
| /llvm-project/mlir/lib/Dialect/Async/Transforms/ |
| H A D | AsyncParallelFor.cpp | 201 coords[i] = b.create<arith::RemSIOp>(index, tripCounts[i]); in delinearize() argument
153 SmallVector<IntegerAttr> tripCounts; global() member
170 ArrayRef<BlockArgument> ParallelComputeFunctionArgs::tripCounts() { tripCounts() function in ParallelComputeFunctionArgs
305 auto tripCounts = values(args.tripCounts(), bounds.tripCounts); createParallelComputeFunction() local
585 doAsyncDispatch(ImplicitLocOpBuilder & b,PatternRewriter & rewriter,ParallelComputeFunction & parallelComputeFunction,scf::ParallelOp op,Value blockSize,Value blockCount,const SmallVector<Value> & tripCounts) doAsyncDispatch() argument
657 doSequentialDispatch(ImplicitLocOpBuilder & b,PatternRewriter & rewriter,ParallelComputeFunction & parallelComputeFunction,scf::ParallelOp op,Value blockSize,Value blockCount,const SmallVector<Value> & tripCounts) doSequentialDispatch() argument
740 SmallVector<Value> tripCounts(op.getNumLoops()); matchAndRewrite() local
[all...] |
| /llvm-project/mlir/include/mlir/Dialect/Linalg/Transforms/ |
| H A D | Transforms.h | 809 SmallVector<T> tripCounts; global() member
|
| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Tiling.cpp | 148 __anon638b672c0102(SmallVector<int64_t> tileSizes, SmallVector<int64_t> tripCounts, int64_t range) computeStaticContinuousTileSizes() argument
|