quantize_config.json

{
    "per_channel": true,
    "reduce_range": true,
    "per_model_config": {
        "encoder_model": {
            "op_types": [
                "Transpose",
                "Cast",
                "Range",
                "Sub",
                "Gather",
                "MatMul",
                "Erf",
                "Mul",
                "Shape",
                "Add",
                "Constant",
                "Expand",
                "Reshape",
                "Pow",
                "Equal",
                "ReduceMean",
                "Not",
                "Sqrt",
                "Div",
                "Where",
                "Softmax",
                "Unsqueeze",
                "Less",
                "Slice",
                "Concat",
                "ConstantOfShape"
            ],
            "weight_type": "QInt8"
        },
        "decoder_postnet_and_vocoder": {
            "op_types": [
                "ConvTranspose",
                "Transpose",
                "Sub",
                "Add",
                "Unsqueeze",
                "Constant",
                "Conv",
                "Reshape",
                "Tanh",
                "Squeeze",
                "Div",
                "LeakyRelu"
            ],
            "weight_type": "QUInt8"
        },
        "decoder_model_merged": {
            "op_types": [
                "Transpose",
                "MatMul",
                "Cast",
                "Sub",
                "Gather",
                "Greater",
                "Sigmoid",
                "Erf",
                "Mul",
                "Shape",
                "If",
                "Add",
                "Constant",
                "RandomUniformLike",
                "Expand",
                "Reshape",
                "Pow",
                "Equal",
                "ReduceMean",
                "Div",
                "Relu",
                "Sqrt",
                "Where",
                "Softmax",
                "Unsqueeze",
                "Slice",
                "Clip",
                "Abs",
                "ReduceSum",
                "Concat",
                "ConstantOfShape"
            ],
            "weight_type": "QInt8"
        },
        "decoder_model": {
            "op_types": [
                "Transpose",
                "MatMul",
                "Cast",
                "Sub",
                "Gather",
                "Greater",
                "Sigmoid",
                "Erf",
                "Mul",
                "Shape",
                "Add",
                "Constant",
                "RandomUniformLike",
                "Expand",
                "Reshape",
                "Pow",
                "Equal",
                "ReduceMean",
                "Div",
                "Relu",
                "Sqrt",
                "Where",
                "Softmax",
                "Unsqueeze",
                "Slice",
                "Clip",
                "Abs",
                "ReduceSum",
                "Concat",
                "ConstantOfShape"
            ],
            "weight_type": "QInt8"
        },
        "decoder_with_past_model": {
            "op_types": [
                "Transpose",
                "MatMul",
                "Cast",
                "Sub",
                "Gather",
                "Greater",
                "Sigmoid",
                "Erf",
                "Mul",
                "Shape",
                "Add",
                "Constant",
                "RandomUniformLike",
                "Expand",
                "Reshape",
                "Pow",
                "Equal",
                "ReduceMean",
                "Div",
                "Relu",
                "Sqrt",
                "Where",
                "Softmax",
                "Unsqueeze",
                "Slice",
                "Clip",
                "Abs",
                "ReduceSum",
                "Concat",
                "ConstantOfShape"
            ],
            "weight_type": "QInt8"
        }
    }
}