Home
last modified time | relevance | path

Searched refs:operandMap (Results 1 – 5 of 5) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DDropUnitDims.cpp126 for (AffineMap operandMap : indexingMaps) { in replaceUnitDims() local
128 if (operandMap.getNumSymbols()) in replaceUnitDims()
131 operandMap.replaceDimsAndSymbols(dimReplacements, symReplacements, in replaceUnitDims()
H A DElementwiseOpFusion.cpp119 AffineMap operandMap = std::get<1>(pair); in areElementwiseOpsFusable() local
120 addToCoveredDims(operandMap); in areElementwiseOpsFusable()
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/
H A DLoopUtils.cpp196 BlockAndValueMapping operandMap; in generateShiftedLoop() local
212 operandMap.map(srcIV, ivRemap); in generateShiftedLoop()
214 operandMap.map(srcIV, loopChunkIV); in generateShiftedLoop()
217 bodyBuilder.clone(*op, operandMap); in generateShiftedLoop()
1022 BlockAndValueMapping operandMap; in generateUnrolledLoop() local
1025 operandMap.map(iterArgs, lastYielded); in generateUnrolledLoop()
1031 operandMap.map(forOpIV, ivUnroll); in generateUnrolledLoop()
1036 Operation *clonedOp = builder.clone(*it, operandMap); in generateUnrolledLoop()
2773 BlockAndValueMapping operandMap; in createFullTiles() local
2775 operandMap.map(loopEn.value().getInductionVar(), in createFullTiles()
[all …]
H A DUtils.cpp295 BlockAndValueMapping operandMap; in hoistAffineIfOp() local
307 operandMap.clear(); in hoistAffineIfOp()
311 hoistOverOpClone = b.clone(*hoistOverOp, operandMap); in hoistAffineIfOp()
/llvm-project-15.0.7/mlir/lib/Dialect/SCF/Utils/
H A DUtils.cpp385 BlockAndValueMapping operandMap; in generateUnrolledLoop() local
388 operandMap.map(iterArgs, lastYielded); in generateUnrolledLoop()
394 operandMap.map(forOpIV, ivUnroll); in generateUnrolledLoop()
399 Operation *clonedOp = builder.clone(*it, operandMap); in generateUnrolledLoop()
405 lastYielded[i] = operandMap.lookup(yieldedValues[i]); in generateUnrolledLoop()