Searched refs:srcDim (Results 1 – 3 of 3) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/NVGPU/Transforms/ |
| H A D | OptimizeSharedMemory.cpp | 41 int64_t srcDim, int64_t tgtDim) { in permuteVectorOffset() argument
44 Value src = indices[srcDim]; in permuteVectorOffset()
96 MemRefType memrefTy, int64_t srcDim, in transformIndices() argument
99 permuteVectorOffset(builder, loc, indices, memrefTy, srcDim, tgtDim); in transformIndices()
212 int64_t srcDim = memRefType.getRank() - 2; in optimizeSharedMemoryReadsAndWrites() local
223 memRefType, srcDim, tgtDim); in optimizeSharedMemoryReadsAndWrites()
236 memRefType, srcDim, tgtDim); in optimizeSharedMemoryReadsAndWrites()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/IR/ |
| H A D | VectorOps.cpp | 1704 int64_t srcDim = srcVectorType.getDimSize(r); in isBroadcastableTo() local
1706 if (srcDim != 1 && srcDim != dstDim) { in isBroadcastableTo()
1708 mismatchingDims->first = srcDim; in isBroadcastableTo()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/MemRef/IR/ |
| H A D | MemRefOps.cpp | 1906 for (int64_t srcDim : group) in computeCollapsedType() local
1907 groupSize = groupSize * Wrapper::size(srcType.getDimSize(srcDim)); in computeCollapsedType()
|