Lines Matching defs:unPackOp

894     tensor::UnPackOp unPackOp, tensor::ExpandShapeOp expandOp,
900   SmallVector<int64_t> innerTileSizes = unPackOp.getStaticTiles();
901   ArrayRef<int64_t> innerDimsPos = unPackOp.getInnerDimsPos();
902   ArrayRef<int64_t> outerDimsPerm = unPackOp.getOuterDimsPerm();
949       expandOp.getLoc(), newExpandType, unPackOp.getSource(),
953       rewriter, unPackOp.getLoc(), newExpandOp, unPackOp.getMixedTiles(),
956       unPackOp.getLoc(), newExpandOp.getResult(), emptyOp,
957       projectedInnerDimsPos, unPackOp.getMixedTiles(), newOuterDimsPerm);
971   LogicalResult matchAndRewrite(tensor::UnPackOp unPackOp,
973     Value result = unPackOp.getResult();
979     if (llvm::any_of(unPackOp.getStaticTiles(), [](int64_t size) {
988           return pushDownUnPackOpThroughExpandShape(unPackOp, op, rewriter,
1004     auto unPackOp = operand.get().getDefiningOp<tensor::UnPackOp>();
1005     if (!unPackOp)
1108   // Insert an unPackOp right after the packed generic.