Rate this Page
★ ★ ★ ★ ★

get_embedding_static_quant_module_mappings#

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

Get module mapping, including mapping for embedding QAT

Return type:

dict[Callable, Any]

On this page
Show Source
PyTorch Libraries