Lines Matching refs:ExtractOp

500     Value result = rewriter.create<ExtractOp>(loc, reductionOp.getType(),  in matchAndRewrite()
1005 void vector::ExtractOp::build(OpBuilder &builder, OperationState &result, in build()
1011 void vector::ExtractOp::build(OpBuilder &builder, OperationState &result, in build()
1021 ExtractOp::inferReturnTypes(MLIRContext *, Optional<Location>, in inferReturnTypes()
1025 ExtractOp::Adaptor op(operands, attributes); in inferReturnTypes()
1038 bool ExtractOp::isCompatibleReturnTypes(TypeRange l, TypeRange r) { in isCompatibleReturnTypes()
1051 LogicalResult vector::ExtractOp::verify() { in verify()
1077 static LogicalResult foldExtractOpFromExtractChain(ExtractOp extractOp) { in foldExtractOpFromExtractChain()
1078 if (!extractOp.getVector().getDefiningOp<ExtractOp>()) in foldExtractOpFromExtractChain()
1082 ExtractOp currentOp = extractOp; in foldExtractOpFromExtractChain()
1085 while (ExtractOp nextOp = currentOp.getVector().getDefiningOp<ExtractOp>()) { in foldExtractOpFromExtractChain()
1094 extractOp->setAttr(ExtractOp::getPositionAttrStrName(), in foldExtractOpFromExtractChain()
1109 ExtractFromInsertTransposeChainState(ExtractOp e);
1179 ExtractOp extractOp;
1200 ExtractOp e) in ExtractFromInsertTransposeChainState()
1315 static Value foldExtractFromBroadcast(ExtractOp extractOp) { in foldExtractFromBroadcast()
1343 extractOp->setAttr(ExtractOp::getPositionAttrStrName(), in foldExtractFromBroadcast()
1349 static Value foldExtractFromShapeCast(ExtractOp extractOp) { in foldExtractFromShapeCast()
1401 extractOp->setAttr(ExtractOp::getPositionAttrStrName(), in foldExtractFromShapeCast()
1408 static Value foldExtractFromExtractStrided(ExtractOp extractOp) { in foldExtractFromExtractStrided()
1443 extractOp->setAttr(ExtractOp::getPositionAttrStrName(), in foldExtractFromExtractStrided()
1449 static Value foldExtractStridedOpFromInsertChain(ExtractOp op) { in foldExtractStridedOpFromInsertChain()
1500 op->setAttr(ExtractOp::getPositionAttrStrName(), in foldExtractStridedOpFromInsertChain()
1511 OpFoldResult ExtractOp::fold(ArrayRef<Attribute>) { in fold()
1532 class ExtractOpFromBroadcast final : public OpRewritePattern<ExtractOp> {
1534 using OpRewritePattern<ExtractOp>::OpRewritePattern;
1536 LogicalResult matchAndRewrite(ExtractOp extractOp, in matchAndRewrite()
1562 class ExtractOpConstantFolder final : public OpRewritePattern<ExtractOp> {
1564 using OpRewritePattern<ExtractOp>::OpRewritePattern;
1566 LogicalResult matchAndRewrite(ExtractOp extractOp, in matchAndRewrite()
1586 void ExtractOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns()