Lines Matching defs:collapseShapeOp

167                                   memref::CollapseShapeOp collapseShapeOp,
173   for (ArrayRef<int64_t> groups : collapseShapeOp.getReassociationIndices()) {
184       sizes[i] = collapseShapeOp.getSrcType().getDimSize(groups[i]);
207   if (collapseShapeOp.getReassociationIndices().empty()) {
210         cast<MemRefType>(collapseShapeOp.getViewSource().getType()).getRank();
547   auto collapseShapeOp = getMemRefOperand(loadOp)
550   if (!collapseShapeOp)
566           loadOp.getLoc(), rewriter, collapseShapeOp, indices, sourceIndices)))
571             loadOp, collapseShapeOp.getViewSource(), sourceIndices);
575             loadOp, collapseShapeOp.getViewSource(), sourceIndices,
580             op, op.getType(), collapseShapeOp.getViewSource(), sourceIndices,
585             op, op.getType(), collapseShapeOp.getViewSource(), sourceIndices,
711   auto collapseShapeOp = getMemRefOperand(storeOp)
714   if (!collapseShapeOp)
730           storeOp.getLoc(), rewriter, collapseShapeOp, indices, sourceIndices)))
735             storeOp, op.getValueToStore(), collapseShapeOp.getViewSource(),
740             storeOp, op.getValueToStore(), collapseShapeOp.getViewSource(),
745             op, op.getValueToStore(), collapseShapeOp.getViewSource(),
750             op, collapseShapeOp.getViewSource(), sourceIndices, op.getMask(),