Searched refs:_format_for_tpu_embedding_ragged_tensor_batch (Results 1 – 1 of 1) sorted by relevance
1758 **self._format_for_tpu_embedding_ragged_tensor_batch(enqueue_datas))1766 def _format_for_tpu_embedding_ragged_tensor_batch(self, enqueue_datas): member in TPUEmbedding