We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8eb5baf commit ae03534Copy full SHA for ae03534
mlir/include/mlir/Dialect/Tensor/IR/TensorOps.td
@@ -213,11 +213,11 @@ def Tensor_DimOp : Tensor_Op<"dim", [
213
214
// Return the dynamic dimension of %A.
215
%c1 = arith.constant 1 : index
216
- %y = tensor.dim %A, %c1 : memref<4x?xf32>
+ %y = tensor.dim %A, %c1 : tensor<4x?xf32>
217
218
// 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
+ %x = "tensor.dim"(%A, %c0) : (tensor<4x?xf32>, index) -> index
+ %y = "tensor.dim"(%A, %c1) : (tensor<4x?xf32>, index) -> index
221
```
222
}];
223
0 commit comments