Home
last modified time | relevance | path

Searched refs:RewritePattern (Results 1 – 25 of 47) sorted by relevance

12

/external/llvm-project/mlir/test/lib/Dialect/SPIRV/
DTestAvailability.cpp97 struct ConvertToAtomCmpExchangeWeak : public RewritePattern {
103 struct ConvertToBitReverse : public RewritePattern {
109 struct ConvertToGroupNonUniformBallot : public RewritePattern {
115 struct ConvertToModule : public RewritePattern {
121 struct ConvertToSubgroupBallot : public RewritePattern {
152 : RewritePattern("test.convert_to_atomic_compare_exchange_weak_op", in ConvertToAtomCmpExchangeWeak()
173 : RewritePattern("test.convert_to_bit_reverse_op", {"spv.BitReverse"}, 1, in ConvertToBitReverse()
188 : RewritePattern("test.convert_to_group_non_uniform_ballot_op", in ConvertToGroupNonUniformBallot()
201 : RewritePattern("test.convert_to_module_op", {"spv.module"}, 1, context) {} in ConvertToModule()
213 : RewritePattern("test.convert_to_subgroup_ballot_op", in ConvertToSubgroupBallot()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dtest_infer_shaped_type_pass.cc28 struct InferReturnTypeComponentsPattern : public RewritePattern {
30 : RewritePattern("mhlo_test.get_return_type_components", 1, context) {} in InferReturnTypeComponentsPattern()
66 struct ReifyReturnTypeShapesPattern : public RewritePattern {
68 : RewritePattern("mhlo_test.reify_return_type_shapes", 1, context) {} in ReifyReturnTypeShapesPattern()
/external/llvm-project/mlir/test/lib/Dialect/Tosa/
DTosaTestPasses.cpp31 struct ConvertTosaNegateOp : public RewritePattern {
33 : RewritePattern(tosa::NegateOp::getOperationName(), 1, context) {} in ConvertTosaNegateOp()
101 struct ConvertTosaConv2DOp : public RewritePattern {
103 : RewritePattern(tosa::Conv2DOp::getOperationName(), 1, context) {} in ConvertTosaConv2DOp()
/external/llvm-project/mlir/include/mlir/Rewrite/
DFrozenRewritePatternList.h23 using NativePatternListT = std::vector<std::unique_ptr<RewritePattern>>;
45 std::vector<std::unique_ptr<RewritePattern>> nativePatterns;
DPatternApplicator.h70 DenseMap<OperationName, SmallVector<const RewritePattern *, 2>> patterns;
73 SmallVector<const RewritePattern *, 1> anyOpPatterns;
/external/llvm-project/mlir/include/mlir/IR/
DPatternMatch.h155 class RewritePattern : public Pattern {
157 virtual ~RewritePattern() {} in ~RewritePattern()
194 struct OpRewritePattern : public RewritePattern {
198 : RewritePattern(SourceOp::getOperationName(), benefit, context) {} in RewritePattern() function
574 using NativePatternListT = std::vector<std::unique_ptr<RewritePattern>>;
580 OwningRewritePatternList(std::unique_ptr<RewritePattern> pattern) { in OwningRewritePatternList()
627 OwningRewritePatternList &insert(std::unique_ptr<RewritePattern> pattern) { in insert()
643 std::enable_if_t<std::is_base_of<RewritePattern, T>::value>
/external/llvm-project/mlir/include/mlir/Conversion/LinalgToStandard/
DLinalgToStandard.h33 class LinalgOpToLibraryCallRewrite : public RewritePattern {
36 : RewritePattern(/*benefit=*/1, MatchAnyOpTypeTag()) {} in LinalgOpToLibraryCallRewrite()
/external/llvm-project/mlir/test/lib/Dialect/Test/
DTestPatterns.cpp56 struct FoldingPattern : public RewritePattern {
59 : RewritePattern(TestOpInPlaceFoldAnchor::getOperationName(), in FoldingPattern()
224 struct TestRegionRewriteUndo : public RewritePattern {
226 : RewritePattern("test.region_builder", 1, ctx) {} in TestRegionRewriteUndo()
248 struct TestCreateBlock : public RewritePattern {
250 : RewritePattern("test.create_block", /*benefit=*/1, ctx) {} in TestCreateBlock()
265 struct TestCreateIllegalBlock : public RewritePattern {
267 : RewritePattern("test.create_illegal_block", /*benefit=*/1, ctx) {} in TestCreateIllegalBlock()
438 struct TestNonRootReplacement : public RewritePattern {
440 : RewritePattern("test.replace_non_root", 1, ctx) {} in TestNonRootReplacement()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlower_tf.cc179 class LowerAddNOp : public RewritePattern {
182 : RewritePattern(AddNOp::getOperationName(), in LowerAddNOp()
242 class LowerDynamicStitchOp : public RewritePattern {
245 : RewritePattern( in LowerDynamicStitchOp()
325 class ConvertFakeQuantWithMinMaxVarsOp : public RewritePattern {
328 : RewritePattern( in ConvertFakeQuantWithMinMaxVarsOp()
454 class LowerInvertPermutationOp : public RewritePattern {
457 : RewritePattern( in LowerInvertPermutationOp()
520 class LowerLgammaOp : public RewritePattern {
523 : RewritePattern(LgammaOp::getOperationName(), in LowerLgammaOp()
[all …]
Dcontraction_fusion.cc35 class FuseIntoContractionOp : public RewritePattern {
38 : RewritePattern(PatternBenefit(1), MatchAnyOpTypeTag()) {} in FuseIntoContractionOp()
Dfold_broadcast.cc35 class ConvertResultsBroadcastableShapeOp : public RewritePattern {
38 : RewritePattern(1, MatchAnyOpTypeTag()) {} in ConvertResultsBroadcastableShapeOp()
/external/llvm-project/mlir/lib/Rewrite/
DPatternApplicator.cpp65 auto processPatternList = [&](SmallVectorImpl<const RewritePattern *> &list) { in applyCostModel()
125 MutableArrayRef<const RewritePattern *> opPatterns; in matchAndRewrite()
167 result = static_cast<const RewritePattern *>(bestPattern) in matchAndRewrite()
/external/llvm-project/mlir/include/mlir/Dialect/Linalg/Transforms/
DTransforms.h393 struct LinalgBaseTilingPattern : public RewritePattern {
444 struct LinalgBaseTileAndFusePattern : public RewritePattern {
496 struct LinalgBaseInterchangePattern : public RewritePattern {
527 struct LinalgBasePromotionPattern : public RewritePattern {
557 struct LinalgBaseVectorizationPattern : public RewritePattern {
591 struct LinalgLoweringPattern : public RewritePattern {
595 : RewritePattern(OpTy::getOperationName(), {}, benefit, context),
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dfuse_bias_tf.cc46 struct ConvertTFBiasAddOp : public RewritePattern {
48 : RewritePattern(TF::BiasAddOp::getOperationName(), 1, context) {} in ConvertTFBiasAddOp()
/external/llvm-project/mlir/test/mlir-tblgen/
Drewriter-indexing.td39 // CHECK: struct test1 : public ::mlir::RewritePattern {
45 // CHECK: struct test2 : public ::mlir::RewritePattern {
/external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
DElementwiseToLinalg.cpp29 struct ConvertAnyElementwiseMappableOpOnRankedTensors : public RewritePattern {
31 : RewritePattern(/*benefit=*/1, MatchAnyOpTypeTag()) {} in ConvertAnyElementwiseMappableOpOnRankedTensors()
DTransforms.cpp111 : RewritePattern(opName, {}, benefit, context), marker(marker), in LinalgBaseTilingPattern()
116 : RewritePattern(benefit, MatchAnyOpTypeTag()), marker(marker), in LinalgBaseTilingPattern()
158 : RewritePattern(opName, {}, benefit, context), in LinalgBaseTileAndFusePattern()
246 : RewritePattern(opName, {}, benefit, context), marker(marker), in LinalgBaseInterchangePattern()
272 : RewritePattern(opName, {}, benefit, context), marker(marker), in LinalgBasePromotionPattern()
300 : RewritePattern(opName, {}, benefit, context), marker(marker) {} in LinalgBaseVectorizationPattern()
/external/llvm-project/mlir/include/mlir/Dialect/Vector/
DVectorTransforms.h120 struct UnrollVectorPattern : public RewritePattern {
123 : RewritePattern(/*benefit=*/1, MatchAnyOpTypeTag()), options(options) {} in UnrollVectorPattern()
205 struct VectorTransferFullPartialRewriter : public RewritePattern {
215 : RewritePattern(benefit, MatchAnyOpTypeTag()), options(options), in RewritePattern() function
/external/llvm-project/mlir/include/mlir/Transforms/
DDialectConversion.h314 class ConversionPattern : public RewritePattern {
347 using RewritePattern::RewritePattern;
353 : RewritePattern(rootName, benefit, ctx), typeConverter(&typeConverter) {} in ConversionPattern()
361 : RewritePattern(benefit, tag), typeConverter(&typeConverter) {} in ConversionPattern()
368 using RewritePattern::rewrite;
/external/llvm-project/mlir/docs/
DPatternRewriter.md21 Patterns are defined by inheriting from the `RewritePattern` class. This class
51 rewrite of the IR. A `RewritePattern` can specify this implementation either via
59 class MyPattern : public RewritePattern {
64 : RewritePattern(MyOp::getOperationName(), benefit, context) {}
67 : RewritePattern(benefit, MatchAnyOpTypeTag()) {}
185 class MyPattern : public RewritePattern {
188 : RewritePattern(MyOp::getOperationName(), benefit, context) {}
/external/llvm-project/mlir/lib/IR/
DPatternMatch.cpp60 void RewritePattern::rewrite(Operation *op, PatternRewriter &rewriter) const { in rewrite()
65 LogicalResult RewritePattern::match(Operation *op) const { in match()
70 void RewritePattern::anchor() {} in anchor()
/external/llvm-project/mlir/include/mlir/Conversion/SCFToStandard/
DSCFToStandard.h19 class RewritePattern; variable
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlegalize_tf.cc133 struct ConvertTF##tf_op##Op : public RewritePattern { \
135 : RewritePattern(TF::tf_op##Op::getOperationName(), 1, context) {} \
453 struct LegalizeUnidirectionalSequenceLstm : public RewritePattern {
455 : RewritePattern(kUnidirectionalSequenceLstm, 1, context) {} in LegalizeUnidirectionalSequenceLstm()
520 struct LegalizeUnidirectionalSequenceRnn : public RewritePattern {
522 : RewritePattern(kUnidirectionalSequenceRnn, 1, context) {} in LegalizeUnidirectionalSequenceRnn()
Dprepare_tf.cc293 class ConvertTFConvOp : public RewritePattern {
297 : RewritePattern(TFConvOpType::getOperationName(), 1, context), in ConvertTFConvOp()
521 struct ConvertTFStridedSlice : public RewritePattern {
523 : RewritePattern(TF::StridedSliceOp::getOperationName(), 2, context) {} in ConvertTFStridedSlice()
848 struct ConvertTFBroadcastTo : public RewritePattern {
850 : RewritePattern(TF::BroadcastToOp::getOperationName(), 1, context) {} in ConvertTFBroadcastTo()
979 struct FusedBatchNormV3Pat : public ::mlir::RewritePattern {
981 : ::mlir::RewritePattern( in FusedBatchNormV3Pat()
1355 struct ConvertRfftToRfft2d : public RewritePattern {
1357 : RewritePattern(TF::RFFTOp::getOperationName(), 1, context) {} in ConvertRfftToRfft2d()
/external/llvm-project/mlir/include/mlir/Conversion/VectorToSCF/
DVectorToSCF.h148 struct VectorTransferRewriter : public RewritePattern {

12