Searched refs:stage1Patterns (Results 1 – 6 of 6) sorted by relevance
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | CodegenStrategy.cpp | 34 SmallVector<FrozenRewritePatternList, 4> stage1Patterns; in transform() local
42 stage1Patterns.emplace_back(t->buildRewritePatterns(context, marker)); in transform()
63 linalg::applyStagedPatterns(func, stage1Patterns, std::move(stage2Patterns), in transform()
|
| D | Transforms.cpp | 317 Operation *op, ArrayRef<FrozenRewritePatternList> stage1Patterns, in applyStagedPatterns() argument
322 for (const auto &patterns : stage1Patterns) { in applyStagedPatterns()
|
| /external/llvm-project/mlir/test/lib/Transforms/ |
| D | TestConvVectorization.cpp | 61 SmallVector<OwningRewritePatternList, 4> stage1Patterns; in runOnOperation() local
62 linalg::populateConvVectorizationPatterns(context, stage1Patterns, tileSizes); in runOnOperation()
64 llvm::move(stage1Patterns, std::back_inserter(frozenStage1Patterns)); in runOnOperation()
|
| D | TestLinalgTransforms.cpp | 438 SmallVector<OwningRewritePatternList, 4> stage1Patterns; in applyMatmulToVectorPatterns() local
441 stage1Patterns); in applyMatmulToVectorPatterns()
443 stage1Patterns.emplace_back(std::make_unique<LinalgTilingPattern<MatmulOp>>( in applyMatmulToVectorPatterns()
451 stage1Patterns); in applyMatmulToVectorPatterns()
454 llvm::move(stage1Patterns, std::back_inserter(frozenStage1Patterns)); in applyMatmulToVectorPatterns()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_tf.cc | 793 OwningRewritePatternList stage1Patterns; in runOnFunction() local
795 addPatterns(context, stage1Patterns); in runOnFunction()
797 FrozenRewritePatternList stage1FrozenPatterns(std::move(stage1Patterns)); in runOnFunction()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Linalg/Transforms/ |
| D | Transforms.h | 787 Operation *op, ArrayRef<FrozenRewritePatternList> stage1Patterns,
|