diff options
Diffstat (limited to 'src/TosaSerialize.cpp')
-rw-r--r-- | src/TosaSerialize.cpp | 19 |
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; } |