Lines Matching defs:unPackOp

356 linalg::lowerUnPack(RewriterBase &rewriter, tensor::UnPackOp unPackOp,
358   Location loc = unPackOp->getLoc();
360   rewriter.setInsertionPoint(unPackOp);
362   RankedTensorType packedTensorType = unPackOp.getSourceType();
366   auto destTensorType = cast<RankedTensorType>(unPackOp.getDest().getType());
367   if (lowerUnpadLikeWithExtractSlice && unPackOp.isLikeUnPad()) {
374     sizes.append(tensor::getMixedSizes(rewriter, loc, unPackOp.getDest()));
377         loc, destTensorType, unPackOp.getSource(),
381     rewriter.replaceOp(unPackOp, extractSliceOp->getResults());
391       tensor::getUnPackInverseSrcPerm(unPackOp, packingMetadata);
407       tensor::getMixedSizes(rewriter, loc, unPackOp.getSource());
412       loc, unPackOp.getSource(), emptyOp, packedToStripMinedShapePerm);
441       tensor::getMixedSizes(rewriter, loc, unPackOp.getDest()),
446       loc, extractSliceOp->getResult(0), unPackOp.getDest());
448   // 7. Replace unPackOp by copyOp.
449   rewriter.replaceOp(unPackOp, copyOp->getResults());
735   // Step 3. Maybe transpose unPackOp.