def METHOD_NAME(self, data_loader):
    self.eval()
    with paddle.no_grad():
        for batch_data in data_loader:
            input_ids, token_type_ids = batch_data
            text_embeddings = self.get_pooled_embedding(input_ids, token_type_ids=token_type_ids)
            yield text_embeddings