Searched refs:std_constant_index (Results 1 – 11 of 11) sorted by relevance
| /external/llvm-project/mlir/lib/Dialect/StandardOps/EDSC/ |
| D | Builders.cpp | 28 res.push_back(std_constant_index(shape[idx])); in getMemRefSizes()
36 lbs.push_back(std_constant_index(0)); in MemRefBoundsCapture()
44 lbs.push_back(std_constant_index(0)); in VectorBoundsCapture()
45 ubs.push_back(std_constant_index(s)); in VectorBoundsCapture()
|
| /external/llvm-project/mlir/test/EDSC/ |
| D | builder-api-test.cpp | 79 lb *std_constant_index(3) + ub; in TEST_FUNC() local
80 lb + std_constant_index(3); in TEST_FUNC()
82 ceilDiv(std_constant_index(31) * floorDiv(i + j * std_constant_index(3), in TEST_FUNC()
83 std_constant_index(32)), in TEST_FUNC()
84 std_constant_index(32)); in TEST_FUNC()
220 buildInNewBlock(TypeRange(), [&](ValueRange) { std_constant_index(0); }); in TEST_FUNC()
221 appendToBlock(b, [&](ValueRange) { std_constant_index(1); }); in TEST_FUNC()
546 Value zero = std_constant_index(0), one = std_constant_index(1); in TEST_FUNC()
622 Value zero = std_constant_index(0), one = std_constant_index(1); in TEST_FUNC()
730 Value zero = std_constant_index(0); in TEST_FUNC()
[all …]
|
| /external/llvm-project/mlir/lib/Dialect/GPU/Transforms/ |
| D | MemoryPromotion.cpp | 53 Value zero = std_constant_index(0); in insertCopyLoops()
54 Value one = std_constant_index(1); in insertCopyLoops()
67 [](int64_t step) { return std_constant_index(step); }); in insertCopyLoops()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | Tiling.cpp | 81 Range{std_constant_index(0), shapeSizes[idx], tileSizes[idx]}); in makeTiledLoopRanges()
239 lbs.push_back(isTiled ? ivs[idxIvs++] : (Value)std_constant_index(0)); in makeTiledShapes()
242 subShapeSizes.push_back(size - std_constant_index(1)); in makeTiledShapes()
267 offsets.push_back(std_constant_index(0)); in makeTiledShapes()
269 strides.push_back(std_constant_index(1)); in makeTiledShapes()
280 auto size = closedIntSize + std_constant_index(1); in makeTiledShapes()
302 strides.push_back(std_constant_index(1)); in makeTiledShapes()
491 auto zero = std_constant_index(0); in tileLinalgOpImpl()
|
| D | Vectorization.cpp | 133 SmallVector<Value, 4> indices(memrefType.getRank(), std_constant_index(0)); in transferReadVector()
143 SmallVector<Value, 4> indices(memrefType.getRank(), std_constant_index(0)); in transferWriteVector()
340 Value zero = std_constant_index(0); in vectorizeLinalgOp()
574 SmallVector<Value, 4> zeros(rank, std_constant_index(0)); in matchAndRewrite()
|
| D | Fusion.cpp | 229 loopRanges[i] = Range{std_constant_index(0), in fuse()
231 std_constant_index(1)}; in fuse()
798 auto zero = std_constant_index(0); in tileRootOperation()
|
| D | Loops.cpp | 228 Value zeroIndex = std_constant_index(0); in getPaddedInput()
|
| D | FusionOnTensors.cpp | 626 linearizedIndex * std_constant_index(expandedDimExtent); in fuseWithReshapeByExpansion()
|
| /external/llvm-project/mlir/include/mlir/Dialect/StandardOps/EDSC/ |
| D | Intrinsics.h | 25 using std_constant_index = ValueBuilder<ConstantIndexOp>; variable
|
| /external/llvm-project/mlir/lib/Conversion/VectorToSCF/ |
| D | VectorToSCF.cpp | 169 return std_constant_index(off); in emitLoops()
574 steps.push_back(std_constant_index(step)); in matchAndRewrite()
659 steps.push_back(std_constant_index(step)); in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Dialect/Vector/ |
| D | VectorTransforms.cpp | 2143 Value sum = xferOp.indices()[indicesIdx] + std_constant_index(vectorSize); in createScopedInBoundsCond()
2219 Value one = std_constant_index(1); in createScopedSubViewIntersection()
2265 Value zero = std_constant_index(0); in createScopedFullPartialLinalgCopy()
2319 Value zero = std_constant_index(0); in createScopedFullPartialVectorTransferRead()
2544 std_constant_index(extract.getResultType().getDimSize(pos)); in matchAndRewrite()
2580 std_constant_index(insert.getSourceVectorType().getDimSize(pos)); in matchAndRewrite()
|