def _prepare_mask_lm_loss(self, name: Text) -> None:
    self._prepare_embed_layers(f'{name}_lm_mask')
    self._prepare_embed_layers(f'{name}_golden_token')
    self._prepare_dot_product_loss(f'{name}_mask', scale_loss=False)
def _prepare_mask_lm_loss(self, name: Text) -> None:
    self._prepare_embed_layers(f'{name}_lm_mask')
    self._prepare_embed_layers(f'{name}_golden_token')
    self._prepare_dot_product_loss(f'{name}_mask', scale_loss=False)