Lines Matching defs:unpackOp

152                                        UnPackOp unpackOp) const {
153     auto outerDimsPerm = unpackOp.getOuterDimsPerm();
156           unpackOp,
160     RankedTensorType sourceType = unpackOp.getSourceType();
161     RankedTensorType destType = unpackOp.getDestType();
163       return rewriter.notifyMatchFailure(unpackOp, "expects static shapes");
165     ArrayRef<int64_t> dimsPos = unpackOp.getInnerDimsPos();
168           unpackOp, "expects unpacking on the innermost dimension");
174   LogicalResult matchAndRewrite(UnPackOp unpackOp,
176     RankedTensorType destType = unpackOp.getDestType();
177     if (failed(isUnpackOnInnerMostDim(rewriter, unpackOp)) &&
178         failed(isPackOn1D(rewriter, unpackOp, destType.getShape(),
179                           unpackOp.getStaticTiles())) &&
180         !unpackOp.isLikeUnPad()) {
184     RankedTensorType sourceType = unpackOp.getSourceType();
190         rewriter, unpackOp.getLoc(), unpackOp.getSource(), destType,
192     rewriter.replaceOp(unpackOp, collapsed);
232     auto unpackOp = sliceOp.getSource().getDefiningOp<UnPackOp>();
233     if (!unpackOp)
236     if (sliceOp.getResultType().getRank() != unpackOp.getDestType().getRank()) {
249     Type elementType = unpackOp.getDestType().getElementType();
253         sliceOp, unpackOp.getSource(), output, unpackOp.getInnerDimsPos(),
254         unpackOp.getMixedTiles(), unpackOp.getOuterDimsPerm());