Searched refs:_format_for_tpu_embedding_sparse_tensor_batch (Results 1 – 1 of 1) sorted by relevance
1764 **self._format_for_tpu_embedding_sparse_tensor_batch(enqueue_datas))1806 def _format_for_tpu_embedding_sparse_tensor_batch(self, enqueue_datas): member in TPUEmbedding