1 files changed, 4 insertions, 15 deletions
diff --git a/src/TosaSerialize.cpp b/src/TosaSerialize.cpp
index 5b0d2bd..0d5e044 100644
--- a/src/TosaSerialize.cpp
+++ b/src/TosaSerialize.cpp
@@ -1286,22 +1286,11 @@ TosaSerializationOperatorBuilder::build<mlir::tosa::MulOp>(
   std::string input0_name = GetTensorName(mul_op.getInput1());
   std::string input1_name = GetTensorName(mul_op.getInput2());
   std::string output_name = GetTensorName(mul_op.getOutput());
+  std::string shift_name = GetTensorName(mul_op.getShift());
 
-  std::vector<std::string> operands;
-  if (mul_op.getOutput()
-          .getType()
-          .cast<mlir::TensorType>()
-          .getElementType()
-          .isInteger(32)) {
-    std::string shift_name = GetTensorName(mul_op.getShift());
-    operands = {input0_name, input1_name, shift_name};
-  } else {
-    operands = {input0_name, input1_name};
-  }
-
-  TosaSerializationOperator *tyop =
-      new TosaSerializationOperator(Op_MUL, Attribute_NONE, nullptr, operands,
-                                    std::vector<std::string>{output_name});
+  TosaSerializationOperator *tyop = new TosaSerializationOperator(
+      Op_MUL, Attribute_NONE, nullptr, {input0_name, input1_name, shift_name},
+      std::vector<std::string>{output_name});
 
   return tyop;
 }