Lines Matching refs:funcOp

130 static void applyPatterns(func::FuncOp funcOp) {  in applyPatterns()  argument
131 MLIRContext *ctx = funcOp.getContext(); in applyPatterns()
242 (void)applyPatternsAndFoldGreedily(funcOp, std::move(patterns)); in applyPatterns()
245 funcOp.walk([](LinalgOp op) { in applyPatterns()
402 static void applyVectorTransferForwardingPatterns(func::FuncOp funcOp) { in applyVectorTransferForwardingPatterns() argument
403 RewritePatternSet forwardPattern(funcOp.getContext()); in applyVectorTransferForwardingPatterns()
404 forwardPattern.add<LinalgCopyVTRForwardingPattern>(funcOp.getContext()); in applyVectorTransferForwardingPatterns()
405 forwardPattern.add<LinalgCopyVTWForwardingPattern>(funcOp.getContext()); in applyVectorTransferForwardingPatterns()
406 (void)applyPatternsAndFoldGreedily(funcOp, std::move(forwardPattern)); in applyVectorTransferForwardingPatterns()
409 static void applyLinalgToVectorPatterns(func::FuncOp funcOp) { in applyLinalgToVectorPatterns() argument
410 RewritePatternSet patterns(funcOp.getContext()); in applyLinalgToVectorPatterns()
411 auto *ctx = funcOp.getContext(); in applyLinalgToVectorPatterns()
418 (void)applyPatternsAndFoldGreedily(funcOp, std::move(patterns)); in applyLinalgToVectorPatterns()
421 static void applyPadTensorToGenericPatterns(func::FuncOp funcOp) { in applyPadTensorToGenericPatterns() argument
422 RewritePatternSet patterns(funcOp.getContext()); in applyPadTensorToGenericPatterns()
423 patterns.add<PadOpTransformationPattern>(funcOp.getContext()); in applyPadTensorToGenericPatterns()
424 (void)applyPatternsAndFoldGreedily(funcOp, std::move(patterns)); in applyPadTensorToGenericPatterns()
427 static void applyGeneralizePadTensorPatterns(func::FuncOp funcOp) { in applyGeneralizePadTensorPatterns() argument
428 RewritePatternSet patterns(funcOp.getContext()); in applyGeneralizePadTensorPatterns()
429 patterns.add<GeneralizePadOpPattern>(funcOp.getContext()); in applyGeneralizePadTensorPatterns()
430 (void)applyPatternsAndFoldGreedily(funcOp, std::move(patterns)); in applyGeneralizePadTensorPatterns()
433 static void applyExtractSliceOfPadTensorSwapPattern(func::FuncOp funcOp) { in applyExtractSliceOfPadTensorSwapPattern() argument
434 RewritePatternSet patterns(funcOp.getContext()); in applyExtractSliceOfPadTensorSwapPattern()
435 patterns.add<ExtractSliceOfPadTensorSwapPattern>(funcOp.getContext()); in applyExtractSliceOfPadTensorSwapPattern()
436 (void)applyPatternsAndFoldGreedily(funcOp, std::move(patterns)); in applyExtractSliceOfPadTensorSwapPattern()
439 static void applyTilePattern(func::FuncOp funcOp, const std::string &loopType, in applyTilePattern() argument
443 MLIRContext *context = funcOp.getContext(); in applyTilePattern()
463 (void)applyPatternsAndFoldGreedily(funcOp, std::move(tilingPattern)); in applyTilePattern()
466 static void applySplitReduction(func::FuncOp funcOp) { in applySplitReduction() argument
467 RewritePatternSet patterns(funcOp.getContext()); in applySplitReduction()
476 StringAttr::get(funcOp.getContext(), "SPLIT"))); in applySplitReduction()
477 (void)applyPatternsAndFoldGreedily(funcOp, std::move(patterns)); in applySplitReduction()
480 static void applyBubbleUpExtractSliceOpPattern(func::FuncOp funcOp) { in applyBubbleUpExtractSliceOpPattern() argument
481 RewritePatternSet patterns(funcOp.getContext()); in applyBubbleUpExtractSliceOpPattern()
483 (void)applyPatternsAndFoldGreedily(funcOp, std::move(patterns)); in applyBubbleUpExtractSliceOpPattern()