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