Skip to content

Commit ae03534

Browse files
committed
[mlir][docs] Fix tensor.dim documentation referencing memref type
1 parent 8eb5baf commit ae03534

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

mlir/include/mlir/Dialect/Tensor/IR/TensorOps.td

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -213,11 +213,11 @@ def Tensor_DimOp : Tensor_Op<"dim", [
213213

214214
// Return the dynamic dimension of %A.
215215
%c1 = arith.constant 1 : index
216-
%y = tensor.dim %A, %c1 : memref<4x?xf32>
216+
%y = tensor.dim %A, %c1 : tensor<4x?xf32>
217217

218218
// Equivalent generic form:
219-
%x = "tensor.dim"(%A, %c0) : (memref<4x?xf32>, index) -> index
220-
%y = "tensor.dim"(%A, %c1) : (memref<4x?xf32>, index) -> index
219+
%x = "tensor.dim"(%A, %c0) : (tensor<4x?xf32>, index) -> index
220+
%y = "tensor.dim"(%A, %c1) : (tensor<4x?xf32>, index) -> index
221221
```
222222
}];
223223

0 commit comments

Comments
 (0)