Lines Matching defs:linalgOp

202     auto linalgOp = current.getDefiningOp<LinalgOp>();
203     if (!linalgOp)
206     current = linalgOp.getDpsInitOperand(opResult.getResultNumber())->get();
278     OpBuilder &b, Location loc, ArrayRef<Range> loopRanges, LinalgOp linalgOp,
288   if (!linalgOp.hasPureBufferSemantics())
289     llvm::append_range(iterArgInitValues, linalgOp.getDpsInits());
297         SmallVector<Value> operandValuesToUse = linalgOp->getOperands();
299           operandValuesToUse = linalgOp.getDpsInputs();
321     OpBuilder &b, Location loc, ArrayRef<Range> loopRanges, LinalgOp linalgOp,
328   if (!linalgOp.hasPureBufferSemantics())
329     llvm::append_range(iterArgInitValues, linalgOp.getDpsInits());
346                                               linalgOp->getOperands());
488     OpBuilder &b, Location loc, ArrayRef<Range> loopRanges, LinalgOp linalgOp,
495   if (!linalgOp.hasPureBufferSemantics())
496     llvm::append_range(iterArgInitValues, linalgOp.getDpsInits());
526         bodyBuilderFn(b, loc, ivs, linalgOp->getOperands());
752 computeAllSliceParameters(OpBuilder &builder, Location loc, LinalgOp linalgOp,
770              linalgOp->getNumOperands() &&
775        llvm::zip(linalgOp->getOpOperands(), valuesToTile)) {
778     AffineMap map = linalgOp.getMatchingIndexingMap(&opOperand);
787                                       linalgOp.isDpsInit(&opOperand))) {
804                                    LinalgOp linalgOp, ValueRange valuesToTile,
810       computeAllSliceParameters(builder, loc, linalgOp, valuesToTile, ivs,
825 void offsetIndices(OpBuilder &b, LinalgOp linalgOp,
828   offsetIndices(rewriter, linalgOp, offsets);
831 void offsetIndices(RewriterBase &b, LinalgOp linalgOp,
833   if (!linalgOp.hasIndexSemantics())
836   for (IndexOp indexOp : linalgOp.getBlock()->getOps<IndexOp>()) {