Lines Matching defs:unPackOp

387     auto unPackOp = linalgOp->getOperand(0).getDefiningOp<UnPackOp>();
389     if (!unPackOp)
397     auto outerDimsPerm = unPackOp.getOuterDimsPerm();
398     auto innerDimsPos = unPackOp.getInnerDimsPos();
413         linalgOp, unPackOp.getSource(), linalgOp.getDpsInits()[0],
414         newInnerDimsPosVec, unPackOp.getMixedTiles(), newOuterDimsPermVec);
426   LogicalResult matchAndRewrite(UnPackOp unPackOp,
428     auto linalgOp = unPackOp.getSource().getDefiningOp<linalg::LinalgOp>();
438     if (failed(reifyResultShapes(rewriter, unPackOp, unpackOpResultDims))) {
444     auto outerDimsPerm = unPackOp.getOuterDimsPerm();
445     auto innerDimsPos = unPackOp.getInnerDimsPos();
446     int64_t destRank = unPackOp.getSourceRank() - innerDimsPos.size();
447     auto mixedInnerTilesVec = unPackOp.getMixedTiles();
454           unPackOp,
466         cast<ShapedType>(unPackOp->getResultTypes()[0]).getElementType();
468         unPackOp->getLoc(), unpackOpResultDims[0], elemType);
471         unPackOp, linalgOp->getOperand(0), output, newInnerDimsPosVec,