Lines Matching refs:outputOperand
144 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()
226 write = b.create<vector::TransferWriteOp>(loc, value, outputOperand->get(), in buildVectorWrite()