get_embedding_qat_module_mappings

class torch.ao.quantization.quantization_mappings.get_embedding_qat_module_mappings

Get module mapping for quantization aware training This is includes default values in addition to enabling qat for embeddings.

Return type:

dict[Callable, Any]