Lines Matching refs:patterns
639 static void insert(RewritePatternSet &patterns) {} in insert() argument
645 static void insert(RewritePatternSet &patterns) { in insert() argument
646 OpTy::getCanonicalizationPatterns(patterns, patterns.getContext()); in insert()
647 CanonicalizationPatternList<OpTypes...>::insert(patterns); in insert()
654 RewritePatternSet patterns(ctx); in getLinalgTilingCanonicalizationPatterns() local
655 populateLinalgTilingCanonicalizationPatterns(patterns); in getLinalgTilingCanonicalizationPatterns()
656 return patterns; in getLinalgTilingCanonicalizationPatterns()
660 RewritePatternSet &patterns) { in populateLinalgTilingCanonicalizationPatterns() argument
661 auto *ctx = patterns.getContext(); in populateLinalgTilingCanonicalizationPatterns()
662 AffineApplyOp::getCanonicalizationPatterns(patterns, ctx); in populateLinalgTilingCanonicalizationPatterns()
663 AffineForOp::getCanonicalizationPatterns(patterns, ctx); in populateLinalgTilingCanonicalizationPatterns()
664 AffineMinOp::getCanonicalizationPatterns(patterns, ctx); in populateLinalgTilingCanonicalizationPatterns()
665 AffineMaxOp::getCanonicalizationPatterns(patterns, ctx); in populateLinalgTilingCanonicalizationPatterns()
666 arith::ConstantIndexOp::getCanonicalizationPatterns(patterns, ctx); in populateLinalgTilingCanonicalizationPatterns()
668 memref::SubViewOp::getCanonicalizationPatterns(patterns, ctx); in populateLinalgTilingCanonicalizationPatterns()
669 memref::ViewOp::getCanonicalizationPatterns(patterns, ctx); in populateLinalgTilingCanonicalizationPatterns()
671 scf::ForOp::getCanonicalizationPatterns(patterns, ctx); in populateLinalgTilingCanonicalizationPatterns()
672 scf::ParallelOp::getCanonicalizationPatterns(patterns, ctx); in populateLinalgTilingCanonicalizationPatterns()
674 tensor::CastOp::getCanonicalizationPatterns(patterns, ctx); in populateLinalgTilingCanonicalizationPatterns()
675 tensor::ExtractSliceOp::getCanonicalizationPatterns(patterns, ctx); in populateLinalgTilingCanonicalizationPatterns()
676 tensor::InsertSliceOp::getCanonicalizationPatterns(patterns, ctx); in populateLinalgTilingCanonicalizationPatterns()
678 InitTensorOp::getCanonicalizationPatterns(patterns, ctx); in populateLinalgTilingCanonicalizationPatterns()
679 tensor::PadOp::getCanonicalizationPatterns(patterns, ctx); in populateLinalgTilingCanonicalizationPatterns()
680 ctx->getLoadedDialect<LinalgDialect>()->getCanonicalizationPatterns(patterns); in populateLinalgTilingCanonicalizationPatterns()
685 >::insert(patterns); in populateLinalgTilingCanonicalizationPatterns()
689 static void insertTilingPatterns(RewritePatternSet &patterns, in insertTilingPatterns() argument
691 auto *ctx = patterns.getContext(); in insertTilingPatterns()
697 >::insert(patterns, options, f); in insertTilingPatterns()
698 patterns.add<PadOpTilingPattern>(ctx, options); in insertTilingPatterns()
702 RewritePatternSet &patterns, const LinalgTilingOptions &options) { in populatePadTensorTilingPatterns() argument
703 auto *ctx = patterns.getContext(); in populatePadTensorTilingPatterns()
704 patterns.add<PadOpTilingPattern>(ctx, options); in populatePadTensorTilingPatterns()
709 RewritePatternSet patterns(ctx); in applyExtractSliceOfPadTensorSwapPattern() local
710 patterns.add<ExtractSliceOfPadTensorSwapPattern>(patterns.getContext()); in applyExtractSliceOfPadTensorSwapPattern()
711 (void)applyPatternsAndFoldGreedily(funcOp, std::move(patterns)); in applyExtractSliceOfPadTensorSwapPattern()
736 RewritePatternSet patterns(ctx); in runOnOperation() local
737 insertTilingPatterns(patterns, options); in runOnOperation()
738 scf::populateSCFForLoopCanonicalizationPatterns(patterns); in runOnOperation()
739 (void)applyPatternsAndFoldGreedily(funcOp, std::move(patterns)); in runOnOperation()