@@ -319,7 +319,7 @@ struct BubbleUpExpandThroughParallelCollapse
319
319
// / Note - this pattern could be extended to be a swap pattern between
320
320
// / `tensor.expand_shape` and `tensor.extract_slice`, but is currently
321
321
// / implemented only as a bubble up pattern for `tensor.extract_slice`.
322
- struct BubbleUpExpandShapeThroughExtractSlice
322
+ struct BubbleUpExtractSliceThroughExpandShape
323
323
: public OpRewritePattern<tensor::ExtractSliceOp> {
324
324
using OpRewritePattern<tensor::ExtractSliceOp>::OpRewritePattern;
325
325
@@ -427,7 +427,7 @@ struct BubbleUpExpandShapeThroughExtractSlice
427
427
// / to tensor<15xf32>
428
428
// / ```
429
429
// / But this is not the intended purpose of the transformation.
430
- struct BubbleUpCollapseShapeThroughExtractSlice
430
+ struct BubbleUpExtractSliceThroughCollapseShape
431
431
: public OpRewritePattern<tensor::ExtractSliceOp> {
432
432
using OpRewritePattern<tensor::ExtractSliceOp>::OpRewritePattern;
433
433
@@ -735,6 +735,6 @@ void mlir::tensor::populateBubbleUpExpandShapePatterns(
735
735
736
736
void mlir::tensor::populateBubbleUpExtractSliceOpPatterns (
737
737
RewritePatternSet &patterns) {
738
- patterns.add <BubbleUpExpandShapeThroughExtractSlice ,
739
- BubbleUpCollapseShapeThroughExtractSlice >(patterns.getContext ());
738
+ patterns.add <BubbleUpExtractSliceThroughExpandShape ,
739
+ BubbleUpExtractSliceThroughCollapseShape >(patterns.getContext ());
740
740
}
0 commit comments