Searched refs:stage2Patterns (Results 1 – 6 of 6) sorted by relevance
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | CodegenStrategy.cpp | 46 OwningRewritePatternList stage2Patterns = in transform() local
48 stage2Patterns.insert<AffineMinSCFCanonicalizationPattern>(context); in transform()
63 linalg::applyStagedPatterns(func, stage1Patterns, std::move(stage2Patterns), in transform()
|
| D | Transforms.cpp | 318 const FrozenRewritePatternList &stage2Patterns, in applyStagedPatterns() argument
331 if (failed(applyPatternsAndFoldGreedily(op, stage2Patterns))) { in applyStagedPatterns()
|
| /external/llvm-project/mlir/test/lib/Transforms/ |
| D | TestConvVectorization.cpp | 66 OwningRewritePatternList stage2Patterns = in runOnOperation() local
68 stage2Patterns.insert<linalg::AffineMinSCFCanonicalizationPattern>(context); in runOnOperation()
84 std::move(stage2Patterns), stage3Transforms); in runOnOperation()
|
| D | TestLinalgTransforms.cpp | 455 FrozenRewritePatternList stage2Patterns = in applyMatmulToVectorPatterns() local
457 applyStagedPatterns(funcOp, frozenStage1Patterns, std::move(stage2Patterns)); in applyMatmulToVectorPatterns()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_tf.cc | 803 OwningRewritePatternList stage2Patterns; in runOnFunction() local
805 addPatterns(context, stage2Patterns); in runOnFunction()
807 stage2Patterns.insert<ApplyExplicitBroadcasting<TF::LessEqualOp>, in runOnFunction()
827 FrozenRewritePatternList stage2FrozenPatterns(std::move(stage2Patterns)); in runOnFunction()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Linalg/Transforms/ |
| D | Transforms.h | 788 const FrozenRewritePatternList &stage2Patterns,
|