Lines Matching refs:DimOp

292 void DimOp::build(OpBuilder &builder, OperationState &result, Value source,  in build()
299 Optional<int64_t> DimOp::getConstantIndex() {  in getConstantIndex()
305 LogicalResult DimOp::verify() {  in verify()
324 OpFoldResult DimOp::fold(ArrayRef<Attribute> operands) {  in fold()
381 struct DimOfCastOp : public OpRewritePattern<DimOp> {
382   using OpRewritePattern<DimOp>::OpRewritePattern;
384   LogicalResult matchAndRewrite(DimOp dimOp,  in matchAndRewrite()
390     rewriter.replaceOpWithNewOp<DimOp>(dimOp, newSource, dimOp.getIndex());  in matchAndRewrite()
396 void DimOp::getCanonicalizationPatterns(RewritePatternSet &results,  in getCanonicalizationPatterns()
1442       dim = b.createOrFold<tensor::DimOp>(  in createCanonicalRankReducingExtractSliceOp()
1581         builder.createOrFold<tensor::DimOp>(getLoc(), getDest(), dim);  in reifyResultShapes()
1795       dim = b.createOrFold<tensor::DimOp>(  in createCanonicalRankReducingInsertSliceOp()