diff --git a/mlir/lib/Dialect/Tensor/IR/TensorOps.cpp b/mlir/lib/Dialect/Tensor/IR/TensorOps.cpp
--- a/mlir/lib/Dialect/Tensor/IR/TensorOps.cpp
+++ b/mlir/lib/Dialect/Tensor/IR/TensorOps.cpp
@@ -1288,8 +1288,8 @@
   if (!sourceType.hasStaticShape())
     return {};
 
-  auto resultType = op.result().getType().dyn_cast<ShapedType>();
-  if (!resultType || !resultType.hasStaticShape())
+  auto resultType = op.result().getType().cast<ShapedType>();
+  if (!resultType.hasStaticShape())
     return {};
 
   // Control the size. Sice the way to get a new constant collects each element,