Searched refs:srcIndices (Results 1 – 9 of 9) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Tosa/Transforms/ |
| H A D | TosaFoldConstantTranspose.cpp | 62 SmallVector<uint64_t, 6> srcIndices(inputType.getRank(), 0); in matchAndRewrite() local
65 srcIndices[dim] = totalCount % inputShape[dim]; in matchAndRewrite()
71 dstIndices[dim] = srcIndices[permValues[dim]]; in matchAndRewrite()
77 outputValues[dstLinearIndex] = attrValues[srcIndices]; in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Utils/ |
| H A D | ReshapeOpsUtils.h | 345 auto &srcIndices = std::get<0>(item); in findCollapsingReassociation() local
347 auto srcSubShape = srcShape.slice(srcIndices.front(), srcIndices.size()); in findCollapsingReassociation()
353 composedReassociation.push_back(srcIndices); in findCollapsingReassociation()
368 shape_indices.push_back(srcIndices.front() + index); in findCollapsingReassociation()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | ConstantFold.cpp | 166 SmallVector<SmallVector<uint64_t>> srcIndices( in matchAndRewrite() local
192 srcIndices[i][dim] = indices[inputDims[i][dim]]; in matchAndRewrite()
198 srcLinearIndices[i] = srcIndices[i].front(); in matchAndRewrite()
204 srcLinearIndices[i] * inputShapes[i][dim] + srcIndices[i][dim]; in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/NVGPU/IR/ |
| H A D | NVGPU.td | 174 Variadic<Index>:$srcIndices,
178 $src `[` $srcIndices `]` `,` $dst `[` $dstIndices `]` `,` $numElements
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/ |
| H A D | VectorTransferSplitRewritePatterns.cpp | 217 SmallVector<OpFoldResult> srcIndices = llvm::to_vector<4>(llvm::map_range( in createSubViewIntersection() local
222 loc, isaWrite ? alloc : xferOp.source(), srcIndices, sizes, strides); in createSubViewIntersection()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/IR/ |
| H A D | AffineOps.h | 99 AffineMap srcMap, ValueRange srcIndices, Value destMemRef,
|
| /llvm-project-15.0.7/mlir/lib/Dialect/MemRef/IR/ |
| H A D | MemRefOps.cpp | 979 Value srcMemRef, ValueRange srcIndices, Value destMemRef, in build() argument
984 result.addOperands(srcIndices); in build()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/IR/ |
| H A D | AffineOps.cpp | 1184 ValueRange srcIndices, Value destMemRef, in build() argument
1191 result.addOperands(srcIndices); in build()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/MemRef/IR/ |
| H A D | MemRefOps.td | 548 OpBuilder<(ins "Value":$srcMemRef, "ValueRange":$srcIndices,
|