Fix type in documentation

Reviewed By: bondhugula Differential Revision: https://reviews.llvm.org/D130274
llvm · Jul 21, 2022 · 5609429 · 5609429
1 parent 0ae4669
commit 5609429
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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>
     ```
   }];