.setShape(mlir::cast<ShapedType>(new_bias_type).getShape());
    q_op.getResult().setType(new_q_op_type);
    auto attr = TypeAttr::get(q_op.getResult().getType());
    q_op.setQtypeAttr(attr);

    // Update DequantizeOp's output shape
    auto new_dq_op_type =
        RankedTensorType::Builder(
            mlir::cast<RankedTensorType>(dq_op.getResult().getType()))