Lines Matching defs:reshape

563     RewriterBase &rewriter, ReassociativeReshapeLikeOp reshape,
575   Location origLoc = reshape.getLoc();
576   Value source = reshape.getSrc();
585   MemRefType reshapeType = reshape.getResultType();
610   unsigned idx = 0, endIdx = reshape.getReassociationIndices().size();
613         getReshapedSizes(reshape, rewriter, origSizes, /*groupId=*/idx);
615         reshape, rewriter, origSizes, origStrides, /*groupId=*/idx);
623   assert(((isa<memref::ExpandShapeOp>(reshape) && idx == sourceRank) ||
624           (isa<memref::CollapseShapeOp>(reshape) && idx == reshapeRank)) &&
661   LogicalResult matchAndRewrite(ReassociativeReshapeLikeOp reshape,
665             rewriter, reshape, getReshapedSizes, getReshapedStrides);
667       return rewriter.notifyMatchFailure(reshape,
668                                          "failed to resolve reshape metadata");
672         reshape, reshape.getType(), stridedMetadata->basePtr,