Lines Matching defs:expandShapeOp

64                                 memref::ExpandShapeOp expandShapeOp,
75   MemRefType srcType = expandShapeOp.getSrcType();
80           rewriter.create<memref::DimOp>(loc, expandShapeOp.getSrc(), i)
88       rewriter, loc, expandShapeOp.getResultType(),
89       expandShapeOp.getReassociationIndices(), srcShape);
95   for (ArrayRef<int64_t> groups : expandShapeOp.getReassociationIndices()) {
499   auto expandShapeOp =
502   if (!expandShapeOp)
518           loadOp.getLoc(), rewriter, expandShapeOp, indices, sourceIndices)))
523             loadOp, expandShapeOp.getViewSource(), sourceIndices);
527             loadOp, expandShapeOp.getViewSource(), sourceIndices,
532             op, op.getType(), expandShapeOp.getViewSource(), sourceIndices,
537             op, op.getType(), expandShapeOp.getViewSource(), sourceIndices,
662   auto expandShapeOp =
665   if (!expandShapeOp)
681           storeOp.getLoc(), rewriter, expandShapeOp, indices, sourceIndices)))
686             storeOp, op.getValueToStore(), expandShapeOp.getViewSource(),
691             storeOp, op.getValueToStore(), expandShapeOp.getViewSource(),
696             op, op.getValueToStore(), expandShapeOp.getViewSource(),
701             op, expandShapeOp.getViewSource(), sourceIndices, op.getMask(),