Lines Matching refs:DimOp

75     Value size = b.create<memref::DimOp>(loc, value, index);  in castOrReallocMemRefValue()
139         dynamicDims.push_back(b.create<memref::DimOp>(loc, shapedValue, i));  in populateDynamicDimSizes()
142         dynamicDims.push_back(b.create<tensor::DimOp>(loc, shapedValue, i));  in populateDynamicDimSizes()
329 struct FoldDimOfAllocTensorOp : public OpRewritePattern<tensor::DimOp> {
330   using OpRewritePattern<tensor::DimOp>::OpRewritePattern;
332   LogicalResult matchAndRewrite(tensor::DimOp dimOp,  in matchAndRewrite()
414     return b.create<tensor::DimOp>(getLoc(), getCopy(), idx);  in getDynamicSize()
543 struct DimOfToTensorFolder : public OpRewritePattern<tensor::DimOp> {
544   using OpRewritePattern<tensor::DimOp>::OpRewritePattern;
546   LogicalResult matchAndRewrite(tensor::DimOp dimOp,  in matchAndRewrite()
552     rewriter.replaceOpWithNewOp<memref::DimOp>(  in matchAndRewrite()
630 struct DimOfCastOp : public OpRewritePattern<memref::DimOp> {
631   using OpRewritePattern<memref::DimOp>::OpRewritePattern;
633   LogicalResult matchAndRewrite(memref::DimOp dimOp,  in matchAndRewrite()
639     rewriter.replaceOpWithNewOp<tensor::DimOp>(dimOp, newSource,  in matchAndRewrite()