Lines Matching refs:OpRewritePattern
195 struct ChainedTensorCast : public OpRewritePattern<CastOp> {
196 using OpRewritePattern<CastOp>::OpRewritePattern;
244 struct TensorCastExtractSlice : public OpRewritePattern<CastOp> {
245 using OpRewritePattern<CastOp>::OpRewritePattern;
381 struct DimOfCastOp : public OpRewritePattern<DimOp> {
382 using OpRewritePattern<DimOp>::OpRewritePattern;
502 : public OpRewritePattern<tensor::ExtractOp> {
503 using OpRewritePattern<tensor::ExtractOp>::OpRewritePattern;
626 struct StaticTensorGenerate : public OpRewritePattern<GenerateOp> {
627 using OpRewritePattern<GenerateOp>::OpRewritePattern;
682 struct ExtractFromTensorGenerate : public OpRewritePattern<tensor::ExtractOp> {
683 using OpRewritePattern<tensor::ExtractOp>::OpRewritePattern;
712 struct ExtractFromTensorCast : public OpRewritePattern<tensor::ExtractOp> {
713 using OpRewritePattern<tensor::ExtractOp>::OpRewritePattern;
892 struct FoldReshapeWithConstant : OpRewritePattern<TensorReshapeOp> {
893 using OpRewritePattern<TensorReshapeOp>::OpRewritePattern;
911 struct FoldReshapeWithFromElements : OpRewritePattern<TensorReshapeOp> {
912 using OpRewritePattern<TensorReshapeOp>::OpRewritePattern;
1188 class ExtractSliceOpCastFolder final : public OpRewritePattern<ExtractSliceOp> {
1190 using OpRewritePattern<ExtractSliceOp>::OpRewritePattern;
1258 : public OpRewritePattern<ExtractSliceOp> {
1260 using OpRewritePattern<ExtractSliceOp>::OpRewritePattern;
1264 : OpRewritePattern<ExtractSliceOp>(context), in ConstantOpExtractSliceFolder()
1592 : public OpRewritePattern<InsertOpTy> {
1594 using OpRewritePattern<InsertOpTy>::OpRewritePattern;
1657 struct InsertSliceOpCastFolder final : public OpRewritePattern<InsertOpTy> {
1658 using OpRewritePattern<InsertOpTy>::OpRewritePattern;
1730 : public OpRewritePattern<InsertOpTy> {
1731 using OpRewritePattern<InsertOpTy>::OpRewritePattern;
1955 struct FoldStaticZeroPadding : public OpRewritePattern<PadOp> {
1956 using OpRewritePattern<PadOp>::OpRewritePattern;
1972 struct FoldSourceTensorCast : public OpRewritePattern<PadOp> {
1973 using OpRewritePattern<PadOp>::OpRewritePattern;
2009 struct FoldTargetTensorCast : public OpRewritePattern<PadOp> {
2010 using OpRewritePattern<PadOp>::OpRewritePattern;
2071 struct FoldOrthogonalPaddings : public OpRewritePattern<PadOp> {
2072 using OpRewritePattern<PadOp>::OpRewritePattern;
2298 : public OpRewritePattern<ParallelInsertSliceOp> {
2300 using OpRewritePattern<ParallelInsertSliceOp>::OpRewritePattern;