Lines Matching refs:ExtractOp

405 LogicalResult ExtractOp::verify() {  in verify()
414 OpFoldResult ExtractOp::fold(ArrayRef<Attribute> operands) {  in fold()
502     : public OpRewritePattern<tensor::ExtractOp> {
503   using OpRewritePattern<tensor::ExtractOp>::OpRewritePattern;
505   LogicalResult matchAndRewrite(tensor::ExtractOp extract,  in matchAndRewrite()
514     auto newExtract = rewriter.create<tensor::ExtractOp>(  in matchAndRewrite()
682 struct ExtractFromTensorGenerate : public OpRewritePattern<tensor::ExtractOp> {
683   using OpRewritePattern<tensor::ExtractOp>::OpRewritePattern;
685   LogicalResult matchAndRewrite(tensor::ExtractOp extract,  in matchAndRewrite()
712 struct ExtractFromTensorCast : public OpRewritePattern<tensor::ExtractOp> {
713   using OpRewritePattern<tensor::ExtractOp>::OpRewritePattern;
715   LogicalResult matchAndRewrite(tensor::ExtractOp extract,  in matchAndRewrite()
721     rewriter.replaceOpWithNewOp<tensor::ExtractOp>(  in matchAndRewrite()