parameters: ()
return_statement: return mapping
docstring: Get module mapping for quantization aware training
          This is includes default values in addition to
          enabling qat for embeddings.
docstring_summary: Get module mapping for quantization aware training
          This is includes default values in addition to
          enabling qat for embeddings.
docstring_tokens: Get module mapping for quantization aware training This is includes default values in addition to enabling qat for embeddings .
function: def get_embedding_qat_module_mappings() -> Dict[Callable, Any]:
      ''' Get module mapping for quantization aware training
          This is includes default values in addition to
          enabling qat for embeddings.
      '''
      mapping = copy.deepcopy(DEFAULT_QAT_MODULE_MAPPINGS)
      mapping[nn.EmbeddingBag] = nnqat.EmbeddingBag
      mapping[nn.Embedding] = nnqat.Embedding
      return mapping
function_tokens: def get_embedding_qat_module_mappings ( ) -> Dict [ Callable , Any ] : mapping = copy . deepcopy ( DEFAULT_QAT_MODULE_MAPPINGS ) mapping [ nn . EmbeddingBag ] = nnqat . EmbeddingBag mapping [ nn . Embedding ] = nnqat . Embedding return mapping