Lines Matching defs:linalgOp

173   auto linalgOp = cast<linalg::LinalgOp>(current);
176     if (!matchReduction(linalgOp.getRegionOutputArgs(), *position,
186     Block &body = linalgOp->getRegion(0).front();
187     if (body.getTerminator()->getOperands() != linalgOp.getRegionInputArgs()) {
193     if (!isElementwise(linalgOp))
198     Block &body = linalgOp->getRegion(0).front();
348   auto linalgOp = cast<linalg::LinalgOp>(current);
350   DiagnosedSilenceableFailure diag = getDimensionsFor(linalgOp, dimensions);
358       linalgOp.getParallelDims(reference);
360       linalgOp.getReductionDims(reference);
374   SmallVector<int64_t, 4> ranges = linalgOp.getStaticLoopRanges();
438   auto linalgOp = cast<linalg::LinalgOp>(current);
440   DiagnosedSilenceableFailure diag = getPositionsFor(linalgOp, positions);
448         linalgOp.getMatchingIndexingMap(linalgOp.getDpsInputOperand(position));
468     Value operand = linalgOp.getDpsInputOperand(position)->get();
529   auto linalgOp = cast<linalg::LinalgOp>(current);
531   DiagnosedSilenceableFailure diag = getPositionsFor(linalgOp, positions);
539         linalgOp.getMatchingIndexingMap(linalgOp.getDpsInitOperand(position));
558     Value operand = linalgOp.getDpsInitOperand(position)->get();
603   auto linalgOp = cast<linalg::LinalgOp>(current);
605       Builder(current).getI64IntegerAttr(linalgOp.getNumDpsInputs());
618   auto linalgOp = cast<linalg::LinalgOp>(current);
620       Builder(current).getI64IntegerAttr(linalgOp.getNumDpsInits());
632   auto linalgOp = cast<linalg::LinalgOp>(current);
633   int64_t numLoops = linalgOp.getNumLoops();
634   Attribute attr = Builder(linalgOp->getContext()).getI64IntegerAttr(numLoops);
646   auto linalgOp = cast<linalg::LinalgOp>(op);
648   DiagnosedSilenceableFailure diag = getPositionFor(linalgOp, position);
652   Value result = linalgOp.getTiedOpResult(linalgOp.getDpsInitOperand(position));