diff --git a/externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/Tcp/IR/TcpOps.td b/externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/Tcp/IR/TcpOps.td
index 409febe70cbb..ed26382d11dd 100644
--- a/externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/Tcp/IR/TcpOps.td
+++ b/externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/Tcp/IR/TcpOps.td
@@ -129,12 +129,12 @@ def Tcp_MulOp : Tcp_BinaryElementwiseOp<"mul", [SameOperandsAndResultElementType
   }];
 
   let arguments = (ins
-    Tcp_Tensor:$in1,
-    Tcp_Tensor:$in2
+    Tcp_FloatOrIntTensor:$in1,
+    Tcp_FloatOrIntTensor:$in2
   );
 
   let results = (outs
-    Tcp_Tensor:$out
+    Tcp_FloatOrIntTensor:$out
   );
 
   let assemblyFormat = "$in1 `,` $in2 attr-dict `:` type($in1) `,` type($in2) `->` type($out)";