Home
last modified time | relevance | path

Searched refs:idMap (Results 1 – 2 of 2) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/
H A DTensorTilingInterfaceImpl.cpp101 auto idMap = AffineMap::getMultiDimIdentityMap(2, b.getContext()); in bubbleUpPadSlice() local
103 return b.createOrFold<AffineMinOp>(loc, idMap, ValueRange{v1, v2}); in bubbleUpPadSlice()
107 return b.createOrFold<AffineMaxOp>(loc, idMap, ValueRange{v1, v2}); in bubbleUpPadSlice()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DSplitReduction.cpp251 AffineMap idMap = in scaleReductionDim() local
253 AffineMap shiftedIdMap = idMap.shiftDims(1, /*offset=*/reductionDimPos + 1); in scaleReductionDim()
264 AffineMap idMap = in insertParallelDim() local
266 AffineMap shiftedIdMap = idMap.shiftDims(1, /*offset=*/reductionDimPos + 1); in insertParallelDim()