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());