@@ -319,7 +319,7 @@ struct BubbleUpExpandThroughParallelCollapse
319319// / Note - this pattern could be extended to be a swap pattern between
320320// / `tensor.expand_shape` and `tensor.extract_slice`, but is currently
321321// / implemented only as a bubble up pattern for `tensor.extract_slice`.
322- struct BubbleUpExpandShapeThroughExtractSlice
322+ struct BubbleUpExtractSliceThroughExpandShape
323323 : public OpRewritePattern<tensor::ExtractSliceOp> {
324324 using OpRewritePattern<tensor::ExtractSliceOp>::OpRewritePattern;
325325
@@ -427,7 +427,7 @@ struct BubbleUpExpandShapeThroughExtractSlice
427427// / to tensor<15xf32>
428428// / ```
429429// / But this is not the intended purpose of the transformation.
430- struct BubbleUpCollapseShapeThroughExtractSlice
430+ struct BubbleUpExtractSliceThroughCollapseShape
431431 : public OpRewritePattern<tensor::ExtractSliceOp> {
432432 using OpRewritePattern<tensor::ExtractSliceOp>::OpRewritePattern;
433433
@@ -735,6 +735,6 @@ void mlir::tensor::populateBubbleUpExpandShapePatterns(
735735
736736void mlir::tensor::populateBubbleUpExtractSliceOpPatterns (
737737 RewritePatternSet &patterns) {
738- patterns.add <BubbleUpExpandShapeThroughExtractSlice ,
739- BubbleUpCollapseShapeThroughExtractSlice >(patterns.getContext ());
738+ patterns.add <BubbleUpExtractSliceThroughExpandShape ,
739+ BubbleUpExtractSliceThroughCollapseShape >(patterns.getContext ());
740740}
0 commit comments