diff --git a/mlir/include/mlir/Dialect/Vector/IR/VectorOps.td b/mlir/include/mlir/Dialect/Vector/IR/VectorOps.td index 2019eb5a9fd7f1..39ad03c8011409 100644 --- a/mlir/include/mlir/Dialect/Vector/IR/VectorOps.td +++ b/mlir/include/mlir/Dialect/Vector/IR/VectorOps.td @@ -300,9 +300,9 @@ def Vector_MultiDimReductionOp : ```mlir %1 = vector.multi_reduction , %0, %acc0 [1, 3] : - vector<4x8x16x32xf32> into vector<4x16xf32> + vector<4x8x16x32xf32> to vector<4x16xf32> %2 = vector.multi_reduction , %1, %acc1 [0, 1] : - vector<4x16xf32> into f32 + vector<4x16xf32> to f32 ``` }]; let builders = [