Lines Matching defs:unpackOp

1245     tensor::UnPackOp unpackOp, PatternRewriter &rewriter) const {
1246   int64_t srcRank = unpackOp.getSourceRank();
1247   int64_t destRank = unpackOp.getDestRank();
1248   ArrayRef<int64_t> srcShape = unpackOp.getSourceType().getShape();
1249   ArrayRef<int64_t> innerDimsPos = unpackOp.getInnerDimsPos();
1250   if (llvm::any_of(unpackOp.getTiledOuterDims(),
1253         unpackOp,
1259   Location loc = unpackOp.getLoc();
1260   Value source = unpackOp.getSource();
1262       unpackOp.getDimAndTileMapping();
1319   auto mixedTiles = unpackOp.getMixedTiles();
1328   Type elemType = unpackOp.getSourceType().getElementType();
1331       loc, readType, unpackOp.getSource(), extractSliceOffsets,
1336       srcShape.take_front(destRank), innerDimsPos, unpackOp.getOuterDimsPerm());
1352   ArrayRef<int64_t> destShape = unpackOp.getDestType().getShape();
1356           tensor::getMixedSize(rewriter, loc, unpackOp.getDest(), i));
1373       loc, partialTile, unpackOp.getDest(), writeOffsets, writeSizes,
1375   rewriter.replaceOp(unpackOp, insert.getResult());