未验证 提交 7a212593 编写于 作者: W Wangzheee 提交者: GitHub

fix delete_quant_dequant_op_pass (#44046)

上级 30846bc9
...@@ -89,6 +89,7 @@ void DeleteQuantDequantOpPass::ApplyImpl(ir::Graph* graph) const { ...@@ -89,6 +89,7 @@ void DeleteQuantDequantOpPass::ApplyImpl(ir::Graph* graph) const {
std::string quantized_op_type = op_desc->Type(); std::string quantized_op_type = op_desc->Type();
op_desc->SetAttr("Input_scale", input_scale); op_desc->SetAttr("Input_scale", input_scale);
op_desc->SetAttr("bit_length", bit_length); op_desc->SetAttr("bit_length", bit_length);
op_desc->SetAttr("enable_int8", true);
op_desc->RenameInput(quant_dequant_output_name, input_name); op_desc->RenameInput(quant_dequant_output_name, input_name);
op_desc->Flush(); op_desc->Flush();
IR_NODE_LINK_TO(input, quantized_node); IR_NODE_LINK_TO(input, quantized_node);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册