Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DLoops.cpp143 for (OpOperand *outputOperand : linalgOp.getOutputOperands()) { in emitScalarImplementation()
145 b, loc, linalgOp.getTiedIndexingMap(outputOperand), allIvsPlusDims); in emitScalarImplementation()
147 b.create<LoadOpTy>(loc, outputOperand->get(), indexing)); in emitScalarImplementation()
155 for (OpOperand *outputOperand : linalgOp.getOutputBufferOperands()) { in emitScalarImplementation()
157 b, loc, linalgOp.getTiedIndexingMap(outputOperand), allIvsPlusDims)); in emitScalarImplementation()
158 outputBuffers.push_back(outputOperand->get()); in emitScalarImplementation()
H A DVectorization.cpp144 static Operation *matchLinalgReduction(OpOperand *outputOperand) { in matchLinalgReduction() argument
145 auto linalgOp = cast<LinalgOp>(outputOperand->getOwner()); in matchLinalgReduction()
147 outputOperand->getOperandNumber() - linalgOp.getNumInputs(); in matchLinalgReduction()
202 OpOperand *outputOperand) { in buildVectorWrite() argument
205 auto linalgOp = cast<LinalgOp>(outputOperand->getOwner()); in buildVectorWrite()
206 ArrayRef<int64_t> shape = linalgOp.getShape(outputOperand); in buildVectorWrite()
208 shape, getElementTypeOrSelf(outputOperand->get().getType())); in buildVectorWrite()
212 reindexIndexingMap(linalgOp.getTiedIndexingMap(outputOperand)); in buildVectorWrite()
217 SmallVector<Value> indices(linalgOp.getRank(outputOperand), in buildVectorWrite()
220 write = b.create<vector::TransferWriteOp>(loc, value, outputOperand->get(), in buildVectorWrite()
[all …]
H A DFusionOnTensors.cpp176 OpOperand *outputOperand = in getTiledProducer() local
178 iterArg->set(outputOperand->get()); in getTiledProducer()
179 tiledOperands[outputOperand->getOperandNumber()] = sliceOp.getResult(); in getTiledProducer()
H A DElementwiseOpFusion.cpp1548 for (OpOperand *outputOperand : genericOp.getOutputOperands()) in matchAndRewrite()
1549 fusedIndexMaps.push_back(genericOp.getTiedIndexingMap(outputOperand)); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/
H A DLinalgOps.cpp1496 OpOperand *outputOperand = in verifyYield() local
1498 Type elementType = getElementTypeOrSelf(outputOperand->get().getType()); in verifyYield()