quantize_config.json

{
    "per_channel": false,
    "reduce_range": false,
    "per_model_config": {
        "decoder_model": {
            "op_types": [
                "Sqrt",
                "MatMul",
                "Concat",
                "Mul",
                "Constant",
                "Shape",
                "Range",
                "Pow",
                "Div",
                "Split",
                "Sub",
                "Where",
                "Reshape",
                "Gather",
                "Unsqueeze",
                "Cast",
                "ReduceMean",
                "Softmax",
                "ConstantOfShape",
                "Slice",
                "Gemm",
                "Tanh",
                "Transpose",
                "Add",
                "Squeeze"
            ],
            "weight_type": "QInt8"
        },
        "decoder_model_merged": {
            "op_types": [
                "Sqrt",
                "MatMul",
                "Concat",
                "Mul",
                "If",
                "Constant",
                "Shape",
                "Range",
                "Pow",
                "Div",
                "Split",
                "Sub",
                "Where",
                "Reshape",
                "Gather",
                "Unsqueeze",
                "Cast",
                "ReduceMean",
                "Softmax",
                "ConstantOfShape",
                "Slice",
                "Gemm",
                "Tanh",
                "Transpose",
                "Add",
                "Squeeze"
            ],
            "weight_type": "QInt8"
        },
        "decoder_with_past_model": {
            "op_types": [
                "Sqrt",
                "MatMul",
                "Concat",
                "Mul",
                "Constant",
                "Shape",
                "Range",
                "Pow",
                "Div",
                "Split",
                "Sub",
                "Where",
                "Reshape",
                "Gather",
                "Unsqueeze",
                "Cast",
                "ReduceMean",
                "Softmax",
                "ConstantOfShape",
                "Slice",
                "Gemm",
                "Tanh",
                "Transpose",
                "Squeeze",
                "Add"
            ],
            "weight_type": "QInt8"
        },
        "encoder_model": {
            "op_types": [
                "Sqrt",
                "MatMul",
                "Concat",
                "Mul",
                "Constant",
                "Expand",
                "Shape",
                "Pow",
                "Div",
                "Conv",
                "Sub",
                "Where",
                "Reshape",
                "Gather",
                "Erf",
                "Unsqueeze",
                "Equal",
                "ReduceMean",
                "Softmax",
                "ConstantOfShape",
                "Slice",
                "Transpose",
                "Add"
            ],
            "weight_type": "QUInt8"
        }
    }
}