Home
last modified time | relevance | path

Searched defs:srcDim (Results 1 – 10 of 10) sorted by relevance

/llvm-project/mlir/lib/Dialect/NVGPU/Transforms/
H A DOptimizeSharedMemory.cpp49 int64_t srcDim, int64_t tgtDim) { in permuteVectorOffset() argument
104 MemRefType memrefTy, int64_t srcDim, in transformIndices() argument
187 int64_t srcDim = memRefType.getRank() - 2; optimizeSharedMemoryReadsAndWrites() local
/llvm-project/mlir/lib/Dialect/Tensor/Transforms/
H A DFoldTensorSubsetOps.cpp179 int64_t srcDim = 0; matchAndRewrite() local
H A DBufferizableOpInterfaceImpl.cpp780 Value srcDim = rewriter.create<tensor::DimOp>(loc, padOp.getSource(), i); bufferize() local
/llvm-project/mlir/lib/Dialect/Affine/Utils/
H A DViewLikeInterfaceUtils.cpp119 for (int64_t srcDim = 0; srcDim < srcRank; ++srcDim) { in resolveSizesIntoOpWithSizes() local
/llvm-project/mlir/lib/Dialect/Tensor/Utils/
H A DUtils.cpp133 int64_t srcDim = 0; isCastLikeInsertSliceOp() local
/llvm-project/flang/runtime/
H A Dnamelist.cpp385 const auto &srcDim{source.GetDimension(j)}; in HandleComponent() local
/llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/Utils/
H A DCodegenUtils.cpp244 auto srcDim = srcShape[i]; in genReshapeDstShape() local
/llvm-project/mlir/lib/Dialect/MemRef/IR/
H A DMemRefOps.cpp2475 for (int64_t srcDim : group) computeCollapsedType() local
/llvm-project/mlir/lib/Dialect/Tensor/IR/
H A DTensorOps.cpp1909 int64_t srcDim = expandShapeOp.getCorrespondingSourceDim(*dim); matchAndRewrite() local
/llvm-project/mlir/lib/Dialect/Vector/IR/
H A DVectorOps.cpp2395 int64_t srcDim = srcVectorType.getDimSize(r); isBroadcastableTo() local