Rate this Page

get_default_qat_module_mappings#

class torch.ao.quantization.quantization_mappings.get_default_qat_module_mappings[source]#

Get default module mapping for quantization aware training

Return type:

dict[Callable, Any]