Lines Matching refs:shapeOp
791 auto *shapeOp = shape.getDefiningOp(); in getAdjustedExtents() local
792 if (auto s = mlir::dyn_cast_or_null<ShapeOp>(shapeOp)) { in getAdjustedExtents()
795 } else if (auto s = mlir::dyn_cast_or_null<ShapeShiftOp>(shapeOp)) { in getAdjustedExtents()
943 mlir::Value dst, mlir::Value src, mlir::Value shapeOp, in genArrayCopy() argument
949 getAdjustedExtents(loc, rewriter, arrLoad, extents, shapeOp); in genArrayCopy()
967 loc, getEleTy(src.getType()), src, shapeOp, in genArrayCopy()
969 factory::originateIndices(loc, rewriter, src.getType(), shapeOp, indices), in genArrayCopy()
972 loc, getEleTy(dst.getType()), dst, shapeOp, in genArrayCopy()
974 factory::originateIndices(loc, rewriter, dst.getType(), shapeOp, indices), in genArrayCopy()
1104 auto shapeOp = getOrReadExtentsAndShapeOp(loc, rewriter, load, extents, in referenceToClone() local
1107 allocateArrayTemp(loc, rewriter, load, extents, shapeOp); in referenceToClone()
1109 load.getMemref(), shapeOp, load.getSlice(), in referenceToClone()
1114 rewriter, loc, getEleTy(load.getType()), eleTy, allocmem, shapeOp, in referenceToClone()
1123 allocmem, shapeOp, store.getSlice(), load); in referenceToClone()
1152 auto shapeOp = getOrReadExtentsAndShapeOp(loc, rewriter, load, extents, in materializeAssignment() local
1155 allocateArrayTemp(loc, rewriter, load, extents, shapeOp); in materializeAssignment()
1158 load.getMemref(), shapeOp, load.getSlice(), in materializeAssignment()
1163 shapeOp, copyUsingSlice ? mlir::Value{} : load.getSlice(), in materializeAssignment()
1172 store.getMemref(), allocmem, shapeOp, in materializeAssignment()