(self, position_ids)
| 784 | self._top_query_embeddings_key = 'top_query_embeddings' |
| 785 | |
| 786 | def forward(self, position_ids): |
| 787 | # Embeddings. |
| 788 | embeddings = self.top_query_embeddings(position_ids) |
| 789 | |
| 790 | return embeddings |
| 791 | |
| 792 | def state_dict_for_save_checkpoint(self, destination=None, prefix='', |
| 793 | keep_vars=False): |
nothing calls this directly
no outgoing calls
no test coverage detected