Lines Matching defs:linalgOp

276     LinalgOp linalgOp = llvm::cast<LinalgOp>(op);
277     SmallVector<AffineMap> res = linalgOp.getIndexingMapsArray();
281     for (int64_t i = 0; i < linalgOp.getNumDpsInits(); ++i) {
282       res.push_back(res[linalgOp.getDpsInitOperand(i)->getOperandNumber()]);
290     LinalgOp linalgOp = llvm::cast<LinalgOp>(op);
292         linalgOp.getIteratorTypesArray();
298     mesh::ReductionKind reductionKind = getReductionKindOfLinalgOp(linalgOp);
308     LinalgOp linalgOp = llvm::cast<LinalgOp>(op);
310     SmallVector<AffineMap> indexingMaps = linalgOp.getIndexingMapsArray();
322         linalgOp.getIteratorTypesArray();
330           linalgOp, spmdizedOperands, operandShardings, resultShardings,