Fix type in documentation

Reviewed By: bondhugula

Differential Revision: https://reviews.llvm.org/D130274

diff --git a/mlir/include/mlir/Dialect/Tensor/IR/TensorOps.td b/mlir/include/mlir/Dialect/Tensor/IR/TensorOps.td
index 7813f88..e3f953f 100644 (file)
--- a/mlir/include/mlir/Dialect/Tensor/IR/TensorOps.td
+++ b/mlir/include/mlir/Dialect/Tensor/IR/TensorOps.td
@@ -1220,7 +1220,7 @@ def Tensor_SplatOp : Tensor_Op<"splat", [
 
     ```mlir
     %s = arith.constant 10.1 : f32
-    %t = tensor.splat %s : tensor<8x16xi32>
+    %t = tensor.splat %s : tensor<8x16xf32>
     ```
 
     TODO: This operation is easy to extend to broadcast to dynamically shaped
@@ -1231,7 +1231,7 @@ def Tensor_SplatOp : Tensor_Op<"splat", [
     // to the sizes of the two dynamic dimensions.
     %m = "foo"() : () -> (index)
     %n = "bar"() : () -> (index)
-    %t = tensor.splat %s [%m, %n] : tensor<?x?xi32>
+    %t = tensor.splat %s [%m, %n] : tensor<?x?xf32>
     ```
   }];