Searched refs:OpRewritePattern (Results 1 – 25 of 86) sorted by relevance
1234
44 class CopyOpToLibraryCallRewrite : public OpRewritePattern<CopyOp> {46 using OpRewritePattern<CopyOp>::OpRewritePattern;54 class CopyTransposeRewrite : public OpRewritePattern<CopyOp> {56 using OpRewritePattern<CopyOp>::OpRewritePattern;64 : public OpRewritePattern<IndexedGenericOp> {66 using OpRewritePattern<IndexedGenericOp>::OpRewritePattern;
298 class AffineMinLowering : public OpRewritePattern<AffineMinOp> {300 using OpRewritePattern<AffineMinOp>::OpRewritePattern;314 class AffineMaxLowering : public OpRewritePattern<AffineMaxOp> {316 using OpRewritePattern<AffineMaxOp>::OpRewritePattern;331 class AffineYieldOpLowering : public OpRewritePattern<AffineYieldOp> {333 using OpRewritePattern<AffineYieldOp>::OpRewritePattern;342 class AffineForLowering : public OpRewritePattern<AffineForOp> {344 using OpRewritePattern<AffineForOp>::OpRewritePattern;362 class AffineParallelLowering : public OpRewritePattern<AffineParallelOp> {364 using OpRewritePattern<AffineParallelOp>::OpRewritePattern;[all …]
256 struct PointwiseExtractPattern : public OpRewritePattern<ExtractMapOp> {261 : OpRewritePattern<ExtractMapOp>(context), filter(constraint) {}293 : public OpRewritePattern<vector::ContractionOp> {295 using OpRewritePattern<vector::ContractionOp>::OpRewritePattern;306 : OpRewritePattern<vector::ContractionOp>(context),334 : public OpRewritePattern<vector::ContractionOp> {336 using OpRewritePattern<vector::ContractionOp>::OpRewritePattern;347 : OpRewritePattern<vector::ContractionOp>(context),378 : public OpRewritePattern<vector::ContractionOp> {380 using OpRewritePattern<vector::ContractionOp>::OpRewritePattern;[all …]
20 : public OpRewritePattern<shape::CstrBroadcastableOp> {22 using OpRewritePattern::OpRewritePattern;31 class RemoveCstrEqOp : public OpRewritePattern<shape::CstrEqOp> {33 using OpRewritePattern::OpRewritePattern;
34 class CompareIConvert : public OpRewritePattern<mhlo::CompareOp> {36 using OpRewritePattern::OpRewritePattern;71 class CompareFConvert : public OpRewritePattern<mhlo::CompareOp> {73 using OpRewritePattern::OpRewritePattern;112 class ConvertIotaOp : public OpRewritePattern<mhlo::IotaOp> {114 using OpRewritePattern::OpRewritePattern;
42 struct DotOpConverter : public OpRewritePattern<DotOp> {43 using OpRewritePattern<DotOp>::OpRewritePattern;102 struct BinaryOpConverter : public OpRewritePattern<LhloOpTy> {103 using OpRewritePattern<LhloOpTy>::OpRewritePattern;
359 struct ConvertConstToTensorConst : public OpRewritePattern<ConstantTensorOp> {360 using OpRewritePattern<ConstantTensorOp>::OpRewritePattern;401 struct RemoveRedundantCast : public OpRewritePattern<CastOp> {402 using OpRewritePattern<CastOp>::OpRewritePattern;436 struct GetTensorShape : public OpRewritePattern<GetShapeOp> {437 using OpRewritePattern<GetShapeOp>::OpRewritePattern;451 struct RemoveRedundantGetElement : public OpRewritePattern<GetElementOp> {452 using OpRewritePattern<GetElementOp>::OpRewritePattern;477 struct RemoveRedundantGetLength : public OpRewritePattern<GetLengthOp> {478 using OpRewritePattern<GetLengthOp>::OpRewritePattern;[all …]
97 struct ForLowering : public OpRewritePattern<ForOp> {98 using OpRewritePattern<ForOp>::OpRewritePattern;190 struct IfLowering : public OpRewritePattern<IfOp> {191 using OpRewritePattern<IfOp>::OpRewritePattern;197 struct ParallelLowering : public OpRewritePattern<mlir::scf::ParallelOp> {198 using OpRewritePattern<mlir::scf::ParallelOp>::OpRewritePattern;263 struct WhileLowering : public OpRewritePattern<WhileOp> {264 using OpRewritePattern<WhileOp>::OpRewritePattern;274 struct DoWhileLowering : public OpRewritePattern<WhileOp> {275 using OpRewritePattern<WhileOp>::OpRewritePattern;
27 : public OpRewritePattern<spirv::GlobalVariableOp> {29 using OpRewritePattern<spirv::GlobalVariableOp>::OpRewritePattern;60 : public OpRewritePattern<spirv::AddressOfOp> {62 using OpRewritePattern<spirv::AddressOfOp>::OpRewritePattern;
166 struct ConvertTosaOp : public OpRewritePattern<OpTy> {167 using OpRewritePattern<OpTy>::OpRewritePattern;192 struct ConvertTosaOp<tosa::MulOp> : public OpRewritePattern<tosa::MulOp> {193 using OpRewritePattern<tosa::MulOp>::OpRewritePattern;221 : public OpRewritePattern<tosa::ArithmeticRightShiftOp> {222 using OpRewritePattern<tosa::ArithmeticRightShiftOp>::OpRewritePattern;
38 struct AtomicRMWOpConverter : public OpRewritePattern<AtomicRMWOp> {40 using OpRewritePattern::OpRewritePattern;75 struct MemRefReshapeOpConverter : public OpRewritePattern<MemRefReshapeOp> {77 using OpRewritePattern::OpRewritePattern;115 struct SignedCeilDivIOpConverter : public OpRewritePattern<SignedCeilDivIOp> {117 using OpRewritePattern::OpRewritePattern;166 struct SignedFloorDivIOpConverter : public OpRewritePattern<SignedFloorDivIOp> {168 using OpRewritePattern::OpRewritePattern;
24 : public OpRewritePattern<shape::CstrBroadcastableOp> {26 using OpRewritePattern::OpRewritePattern;95 class ConvertCstrRequireOp : public OpRewritePattern<shape::CstrRequireOp> {97 using OpRewritePattern::OpRewritePattern;
49 struct UnrollSCFForOp : public OpRewritePattern<scf::ForOp> {50 using OpRewritePattern<scf::ForOp>::OpRewritePattern;106 struct SimplifySCFIfOp : public OpRewritePattern<scf::IfOp> {107 using OpRewritePattern<scf::IfOp>::OpRewritePattern;
308 struct FuseFullyConnectedAndAdd : public OpRewritePattern<TFL::AddOp> {309 using OpRewritePattern<TFL::AddOp>::OpRewritePattern;405 struct FuseFullyConnectedAndReluX : public OpRewritePattern<ReluXOp> {406 using OpRewritePattern<ReluXOp>::OpRewritePattern;435 struct FuseFullyConnectedAndMul : public OpRewritePattern<TFL::MulOp> {436 using OpRewritePattern<TFL::MulOp>::OpRewritePattern;536 struct FuseAffinOpAndMulWithQDQs : public OpRewritePattern<TFL::MulOp> {537 using OpRewritePattern<TFL::MulOp>::OpRewritePattern;615 struct FuseBinaryOpToFollowingAffineOp : public OpRewritePattern<AffineOpType> {616 using OpRewritePattern<AffineOpType>::OpRewritePattern;[all …]
1126 class ConvertBiasAddOp : public OpRewritePattern<TF::BiasAddOp> {1128 using OpRewritePattern::OpRewritePattern;1157 class ConvertConvOp : public OpRewritePattern<OpTy> {1159 using OpRewritePattern<OpTy>::OpRewritePattern;1303 class ConvertBF16FloorDivOp : public OpRewritePattern<TF::FloorDivOp> {1305 using OpRewritePattern::OpRewritePattern;1331 class ConvertBroadcastToOp : public OpRewritePattern<TF::BroadcastToOp> {1333 using OpRewritePattern::OpRewritePattern;1369 class ConvertDiagPartOp : public OpRewritePattern<TF::DiagPartOp> {1371 using OpRewritePattern::OpRewritePattern;[all …]
671 : public OpRewritePattern<vector::TransferReadOp> {672 using OpRewritePattern<vector::TransferReadOp>::OpRewritePattern;698 : public OpRewritePattern<vector::TransferWriteOp> {699 using OpRewritePattern<vector::TransferWriteOp>::OpRewritePattern;718 : public OpRewritePattern<AffineMinOp> {719 using OpRewritePattern<AffineMinOp>::OpRewritePattern;758 class ConvOpVectorization : public OpRewritePattern<ConvOp> {759 using OpRewritePattern<ConvOp>::OpRewritePattern;764 : OpRewritePattern<ConvOp>(context) { in ConvOpVectorization()
28 class LoadOpOfSubViewFolder final : public OpRewritePattern<OpTy> {30 using OpRewritePattern<OpTy>::OpRewritePattern;43 class StoreOpOfSubViewFolder final : public OpRewritePattern<OpTy> {45 using OpRewritePattern<OpTy>::OpRewritePattern;
33 class ConvertTFBatchMatMulOp : public OpRewritePattern<BatchMatMulOpType> {34 using OpRewritePattern<BatchMatMulOpType>::OpRewritePattern;
27 struct FuseParallelMapAndBatch : public OpRewritePattern<BatchDatasetV2Op> {28 using OpRewritePattern<BatchDatasetV2Op>::OpRewritePattern;
47 : public OpRewritePattern<BatchMatMulOpType> {48 using OpRewritePattern<BatchMatMulOpType>::OpRewritePattern;
28 class FakeQuantRewrite : public OpRewritePattern<FakeQuantOp> {30 using OpRewritePattern<FakeQuantOp>::OpRewritePattern;33 : OpRewritePattern<FakeQuantOp>(ctx), hadFailure(hadFailure) {} in FakeQuantRewrite()
133 struct ConstantOpLowering : public OpRewritePattern<toy::ConstantOp> {134 using OpRewritePattern<toy::ConstantOp>::OpRewritePattern;200 struct ReturnOpLowering : public OpRewritePattern<toy::ReturnOp> {201 using OpRewritePattern<toy::ReturnOp>::OpRewritePattern;
133 struct ConstantOpLowering : public OpRewritePattern<toy::ConstantOp> {134 using OpRewritePattern<toy::ConstantOp>::OpRewritePattern;199 struct ReturnOpLowering : public OpRewritePattern<toy::ReturnOp> {200 using OpRewritePattern<toy::ReturnOp>::OpRewritePattern;
687 : public OpRewritePattern<LinalgOpTy> {688 using OpRewritePattern<LinalgOpTy>::OpRewritePattern;748 : public OpRewritePattern<GenericOpTy> {749 using OpRewritePattern<GenericOpTy>::OpRewritePattern;785 : public OpRewritePattern<TensorReshapeOp> {786 using OpRewritePattern<TensorReshapeOp>::OpRewritePattern;843 : public OpRewritePattern<TensorReshapeOp> {844 using OpRewritePattern<TensorReshapeOp>::OpRewritePattern;870 struct FoldSplatConstants : public OpRewritePattern<LinalgOpTy> {871 using OpRewritePattern<LinalgOpTy>::OpRewritePattern;[all …]