quantize_config.json

{
    "per_channel": true,
    "reduce_range": true,
    "per_model_config": {
        "decoder_model": {
            "op_types": [
                "Reshape",
                "Equal",
                "Not",
                "Div",
                "Squeeze",
                "Shape",
                "Transpose",
                "Unsqueeze",
                "Softmax",
                "Cast",
                "Add",
                "Sub",
                "Sqrt",
                "ReduceMean",
                "MatMul",
                "Gather",
                "Concat",
                "Mul",
                "ConstantOfShape",
                "Relu",
                "Identity",
                "Expand",
                "CumSum",
                "Less",
                "Pow",
                "Slice",
                "Range",
                "Where",
                "Constant"
            ],
            "weight_type": "QInt8"
        },
        "encoder_model": {
            "op_types": [
                "Reshape",
                "Equal",
                "Not",
                "Div",
                "Shape",
                "Transpose",
                "Unsqueeze",
                "Softmax",
                "Cast",
                "Add",
                "Sub",
                "Sqrt",
                "ReduceMean",
                "MatMul",
                "Gather",
                "Concat",
                "Mul",
                "ConstantOfShape",
                "Relu",
                "Identity",
                "Expand",
                "CumSum",
                "Pow",
                "Where",
                "Constant"
            ],
            "weight_type": "QInt8"
        },
        "decoder_model_merged": {
            "op_types": [
                "Reshape",
                "Equal",
                "Not",
                "If",
                "Div",
                "Squeeze",
                "Shape",
                "Transpose",
                "Unsqueeze",
                "Softmax",
                "Cast",
                "Add",
                "Sub",
                "Sqrt",
                "ReduceMean",
                "MatMul",
                "Gather",
                "Concat",
                "Mul",
                "ConstantOfShape",
                "Relu",
                "Identity",
                "Expand",
                "CumSum",
                "Less",
                "Pow",
                "Slice",
                "Range",
                "Where",
                "Constant"
            ],
            "weight_type": "QInt8"
        },
        "decoder_with_past_model": {
            "op_types": [
                "Reshape",
                "Equal",
                "Not",
                "Div",
                "Shape",
                "Transpose",
                "Unsqueeze",
                "Softmax",
                "Cast",
                "Add",
                "Sub",
                "Sqrt",
                "ReduceMean",
                "MatMul",
                "Gather",
                "Concat",
                "Mul",
                "ConstantOfShape",
                "Relu",
                "Identity",
                "CumSum",
                "Expand",
                "Pow",
                "Where",
                "Constant"
            ],
            "weight_type": "QInt8"
        }
    }
}